halcon第十五Halcon神經(jīng)網(wǎng)絡(luò)(MLP)金屬零件分類

一、檢測目的
對不同的金屬零件進(jìn)行MLP分類
二、代碼示例

dev_update_off ()
dev_close_window ()
dev_open_window (0, 0, 640, 480, 'black', WindowHandle)
set_display_font (WindowHandle, 16, 'mono', 'true', 'false')
dev_set_colored (6)
dev_set_draw ('margin')
dev_set_line_width (3)

create_class_mlp (6, 5, 3, 'softmax', 'normalization', 3, 42, MLPHandle)

*nuts:螺母、washers:墊圈、retainers:夾持器
FileNames := ['nuts_01','nuts_02','nuts_03','washers_01','washers_02','washers_03','retainers_01','retainers_02','retainers_03']
Classes := [0,0,0,1,1,1,2,2,2]

for J := 0 to |FileNames| - 1 by 1
    read_image (Image, 'rings/' + FileNames[J])
    dev_display (Image)
    dev_set_colored (6)
    
    binary_threshold (Image, Region, 'max_separability', 'dark', UsedThreshold)
    connection (Region, ConnectedRegions)
    fill_up (ConnectedRegions, Regions)
    
    count_obj (Regions, Number)
    for Index := 1 to Number by 1
        select_obj (Regions, ObjectSelected, Index)
        
        select_obj (ObjectSelected, SingleRegion, 1)
        *區(qū)域的圓度(與圓相似)的形狀因子
        circularity (SingleRegion, Circularity)
        *輪廓形狀因子
        roundness (SingleRegion, Distance, Sigma, Roundness, Sides)
        *區(qū)域的幾何矩
        moments_region_central_invar (SingleRegion, PSI1, PSI2, PSI3, PSI4)
        Features := [Circularity,Roundness,PSI1,PSI2,PSI3,PSI4]
        
        add_sample_class_mlp (MLPHandle, Features,  Classes[J])
        
        disp_continue_message (WindowHandle, 'black', 'true')
    endfor
    
endfor

train_class_mlp (MLPHandle, 200, 1, 0.01, Error, ErrorLog)
*清除多層感知器的訓(xùn)練數(shù)據(jù)
clear_samples_class_mlp (MLPHandle)

dev_set_draw ('fill')

for J := 1 to 4 by 1
    read_image (Image, 'rings/mixed_' + J$'02d')
    dev_display (Image)
    dev_set_color ('black')
    dev_set_draw ('margin')
    disp_message (WindowHandle, 'Classifiy Image' + J, 'window', 10, 10, 'black', 'true')
    
    binary_threshold (Image, Region, 'max_separability', 'dark', UsedThreshold)
    connection (Region, ConnectedRegions)
    fill_up (ConnectedRegions, Regions)
    
    count_obj (Regions, Number1)
    Classes := []
    for Index1 := 1 to Number1 by 1
        select_obj (Regions, ObjectSelected1, Index1)
        
        select_obj (ObjectSelected1, SingleRegion, 1)
        circularity (SingleRegion, Circularity)
        roundness (SingleRegion, Distance, Sigma, Roundness, Sides)
        moments_region_central_invar (SingleRegion, PSI1, PSI2, PSI3, PSI4)
        Features := [Circularity,Roundness,PSI1,PSI2,PSI3,PSI4]
        classify_class_mlp (MLPHandle, Features, 1, Class, Confidence)
        Classes := [Classes,Class]
        
    endfor
        count_obj (Regions, Number2)
        Colors := ['yellow','magenta','green']
        for zf := 1 to Number2 by 1
            select_obj (Regions, Region, zf)
            dev_set_color (Colors[Classes[zf - 1]])
            dev_display (Region)
        endfor
            if (J < 4)
        disp_continue_message (WindowHandle, 'black', 'true')
        stop ()
    endif
endfor

stop ()
* 
* Clear the classifier from memory
clear_class_mlp (MLPHandle)
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容