結合模板匹配與測量的剃鬚刀缺陷檢測

星辰_2020發表於2020-10-29

模板匹配與測量

原圖

模板圖

在這裡插入圖片描述

檢測圖

在這裡插入圖片描述

程式碼

* ------------------------------------------------------------------------------------------------
*此示例程式使用基於形狀的匹配來對齊度量的roi
*工具，然後檢查單個刀片。

*該程式有兩種執行模式：（1）全仿射變換
*（2）採用平移測度
*修改下一行以在模式之間切換。

*該引數控制是否使用translate_measure
USING_TRANSLATE_MEASURE := 1
* ------------------------------------------------------------------------------------------------

dev_update_window ('off')
read_image (ModelImage, 'razors1')
get_image_pointer1 (ModelImage, Pointer, Type, Width, Height)
dev_close_window ()
dev_open_window (0, 0, Width, Height, 'white', WindowHandle)
dev_set_part (0, 0, Height - 1, Width - 1)
dev_display (ModelImage)
*顯示設定
dev_set_color ('cyan')
dev_set_draw ('margin')
dev_set_line_width (2)
stop ()
* -------------------  start of the application  ----------------
* 模板物件確定
Row1 := 46
Column1 := 57
Row2 := 79
Column2 := 94
gen_rectangle1 (ROIPart1, Row1, Column1, Row2, Column2)
gen_rectangle1 (ROIPart2, Row1 + 364, Column1 + 13, Row2 + 364, Column2 + 13)
union2 (ROIPart1, ROIPart2, ModelROI)
area_center (ModelROI, Area, CenterROIRow, CenterROIColumn)
dev_display (ModelImage)
dev_display (ModelROI)
stop ()
* 建模
reduce_domain (ModelImage, ModelROI, ImageROI)
create_shape_model (ImageROI, 4, 0, 0, 'auto', 'none', 'use_polarity', 30, 10, ModelID)
inspect_shape_model (ImageROI, ShapeModelImage, ShapeModelRegion, 1, 30)
*通過get_shape_model_contours可知建立的模型中心在（0,0）處
get_shape_model_contours (ShapeModel, ModelID, 1)

dev_clear_window ()
dev_set_color ('blue')
dev_display (ShapeModelRegion)
stop ()
* 測量ROI區域確定
Rect1Row := 244
Rect1Col := 73
DistColRect1Rect2 := 17
Rect2Row := Rect1Row
Rect2Col := Rect1Col + DistColRect1Rect2
RectPhi := rad(90)
RectLength1 := 122
RectLength2 := 2
gen_rectangle2 (MeasureROI1, Rect1Row, Rect1Col, RectPhi, RectLength1, RectLength2)
gen_rectangle2 (MeasureROI2, Rect2Row, Rect2Col, RectPhi, RectLength1, RectLength2)
dev_display (ModelImage)
dev_set_color ('yellow')
dev_display (MeasureROI1)
dev_display (MeasureROI2)
***************************************************
**方法一：建模預設原點仿射變換時，平移模型區域中心到原點
*通過變數OriginalClipRegion保留原始引數屬性，'clip_region'是否裁剪
* get_system ('clip_region', OriginalClipRegion)
* set_system ('clip_region', 'false')
*把建模的區域的中心移到0,0點，此時角度為0。以便後面的仿射變換
* move_region (MeasureROI1, MeasureROI1Ref, -CenterROIRow, -CenterROIColumn)
* move_region (MeasureROI2, MeasureROI2Ref, -CenterROIRow, -CenterROIColumn)
* set_system ('clip_region', OriginalClipRegion)

* DistRect1CenterRow := Rect1Row - CenterROIRow
* DistRect1CenterCol := Rect1Col - CenterROIColumn
* DistRect2CenterRow := Rect2Row - CenterROIRow
* DistRect2CenterCol := Rect2Col - CenterROIColumn
****************************************************
if (USING_TRANSLATE_MEASURE != 0)
   
    *測量物件生成一次，後續可以使用translate_measure繼續使用
    gen_measure_rectangle2 (Rect1Row, Rect1Col, RectPhi, RectLength1, RectLength2, Width, Height, 'bilinear', MeasureHandle1)
    gen_measure_rectangle2 (Rect2Row, Rect2Col, RectPhi, RectLength1, RectLength2, Width, Height, 'bilinear', MeasureHandle2)
endif
stop ()
* find_shape_model模板匹配
read_image (SearchImage, 'razors2')
dev_display (SearchImage)
find_shape_model (SearchImage, ModelID, 0, 0, 0.8, 0, 0.5, 'least_squares', 0, 0.7, RowCheck, ColumnCheck, AngleCheck, Score)
if (|Score| > 0)
    for i := 0 to |Score| - 1 by 1
        * 方法一：建模預設原點仿射變換
        *vector_angle_to_rigid (0, 0, 0, RowCheck[i], ColumnCheck[i], AngleCheck[i], MovementOfObject)
         * affine_trans_contour_xld (ShapeModel, ModelAtNewPosition, MovementOfObject)
         *方法二：建模區域不回原點，使用監視模型  
        vector_angle_to_rigid (CenterROIRow, CenterROIColumn, 0, RowCheck[i], ColumnCheck[i], AngleCheck[i], MovementOfObject)       
        affine_trans_region (ShapeModelRegion, ModelAtNewPosition, MovementOfObject, 'nearest_neighbor')
        dev_display (ModelAtNewPosition)
        
        * 方法二：建模區域不回原點 
        affine_trans_region (MeasureROI1, MeasureROI1AtNewPosition, MovementOfObject, 'constant')
        affine_trans_region (MeasureROI2, MeasureROI2AtNewPosition, MovementOfObject, 'constant')
        *方法一：建模預設原點仿射變換
       * affine_trans_region (MeasureROI1Ref, MeasureROI1AtNewPosition, MovementOfObject, 'constant')
       * affine_trans_region (MeasureROI2Ref, MeasureROI2AtNewPosition, MovementOfObject, 'constant')
        
        dev_display (MeasureROI1AtNewPosition)
        dev_display (MeasureROI2AtNewPosition)
        *方法一：建模預設原點仿射變換
*         affine_trans_pixel (MovementOfObject, DistRect1CenterRow, DistRect1CenterCol, Rect1RowCheck, Rect1ColCheck)
*         affine_trans_pixel (MovementOfObject, DistRect2CenterRow, DistRect2CenterCol, Rect2RowCheck, Rect2ColCheck)
        * 方法二：建模區域不回原點 
        affine_trans_pixel (MovementOfObject, Rect1Row, Rect1Col, Rect1RowCheck, Rect1ColCheck)
        affine_trans_pixel (MovementOfObject, Rect2Row, Rect2Col, Rect2RowCheck, Rect2ColCheck)
        
        
        if (USING_TRANSLATE_MEASURE != 0)
            * -> translate the already created measure objects
            *前面已經生成了測量物件，直接轉換測量
            translate_measure (MeasureHandle1, Rect1RowCheck, Rect1ColCheck)
            translate_measure (MeasureHandle2, Rect2RowCheck, Rect2ColCheck)
            measure_pairs (SearchImage, MeasureHandle1, 2, 25, 'negative', 'all', RowEdge11, ColEdge11, Amp11, RowEdge21, ColEdge21, Amp21, Width1, Distance1)
            measure_pairs (SearchImage, MeasureHandle2, 2, 25, 'negative', 'all', RowEdge12, ColEdge12, Amp12, RowEdge22, ColEdge22, Amp22, Width2, Distance2)
        else
            * -> create new measure objects and destroy them after the measurement
            *每次測量生成測量物件
            RectPhiCheck := RectPhi + AngleCheck[i]
            gen_measure_rectangle2 (Rect1RowCheck, Rect1ColCheck, RectPhiCheck, RectLength1, RectLength2, Width, Height, 'bilinear', MeasureHandle1)
            gen_measure_rectangle2 (Rect2RowCheck, Rect2ColCheck, RectPhiCheck, RectLength1, RectLength2, Width, Height, 'bilinear', MeasureHandle2)
            * step 5: perform the measurement
            measure_pairs (SearchImage, MeasureHandle1, 2, 25, 'negative', 'all', RowEdge11, ColEdge11, Amp11, RowEdge21, ColEdge21, Amp21, Width1, Distance1)
            measure_pairs (SearchImage, MeasureHandle2, 2, 25, 'negative', 'all', RowEdge12, ColEdge12, Amp12, RowEdge22, ColEdge22, Amp22, Width2, Distance2)
            close_measure (MeasureHandle1)
            close_measure (MeasureHandle2)
        endif
        * step 6: check for too short or missing teeth
        NumberTeeth1 := |Width1|
        NumberTeeth2 := |Width2|
        dev_set_color ('red')
        *判斷測量的邊緣對數量是否不到37
        if (NumberTeeth1 < 37)
            for j := 0 to NumberTeeth1 - 2 by 1
                *如果邊緣對寬度大於4，那麼框選該部分並設定該部分割槽域在新視窗顯示
                if (Distance1[j] > 4.0)
                    RowFault := round(0.5 * (RowEdge11[j + 1] + RowEdge21[j]))
                    ColFault := round(0.5 * (ColEdge11[j + 1] + ColEdge21[j]))
                    disp_rectangle2 (WindowHandle, RowFault, ColFault, 0, 4, 4)
                    dev_open_window (0, Width + 20, 80, 80, 'black', WindowHandleZoom)
                    dev_set_part (RowFault - 10, ColFault - 10, RowFault + 10, ColFault + 10)
                    dev_display (SearchImage)
                    disp_rectangle2 (WindowHandleZoom, RowFault, ColFault, 0, 4, 4)
                    stop ()
                    dev_close_window ()
                    dev_set_part (0, 0, Height - 1, Width - 1)
                endif
            endfor
        endif
        if (NumberTeeth2 < 37)
            for j := 0 to NumberTeeth2 - 2 by 1
                if (Distance2[j] > 4.0)
                    RowFault := round(0.5 * (RowEdge12[j + 1] + RowEdge22[j]))
                    ColFault := round(0.5 * (ColEdge12[j + 1] + ColEdge22[j]))
                    disp_rectangle2 (WindowHandle, RowFault, ColFault, 0, 4, 4)
                    dev_open_window (0, Width + 20, 80, 80, 'black', WindowHandleZoom)
                    dev_set_part (RowFault - 10, ColFault - 10, RowFault + 10, ColFault + 10)
                    dev_display (SearchImage)
                    disp_rectangle2 (WindowHandleZoom, RowFault, ColFault, 0, 4, 4)
                    stop ()
                    dev_close_window ()
                    dev_set_part (0, 0, Height - 1, Width - 1)
                endif
            endfor
        endif
        dev_set_color ('yellow')
        stop ()
    endfor
endif
* -------------------  end of the application  -----------------
* clean up
if (USING_TRANSLATE_MEASURE != 0)
    close_measure (MeasureHandle1)
    close_measure (MeasureHandle2)
endif
dev_update_window ('on')
clear_shape_model (ModelID)
dump_window (WindowHandle, 'png', '識別檢測結果')

結果

在這裡插入圖片描述

相關文章