HALCON示例程序classify_image_class_gmm.hdev使用GMM分類器分割彩色圖片

示例程序源碼（加注釋）

dev_update_off ()
dev_close_window ()
dev_open_window (0, 0, 735, 485, ‘black’, WindowHandle)
set_display_font (WindowHandle, 14, ‘mono’, ‘true’, ‘false’)
dev_set_draw (‘margin’)
dev_set_colored (6)
dev_set_line_width (3)
read_image (Image, ‘patras’)
dev_display (Image)

• 定義顏色數組
  Color := [‘indian red’,‘cornflower blue’,‘white’,‘black’,‘yellow’]

• 為要進行分類的四個類創建矩形框并顯示
  gen_rectangle1 (Sea, 10, 10, 120, 270)
  gen_rectangle2 (Deck, [170,400], [350,375], [-0.56,-0.75], [64,104], [26,11])
  union1 (Deck, Deck)
  gen_rectangle1 (Walls, 355, 623, 420, 702)
  gen_rectangle2 (Chimney, 286, 623, -0.56, 64, 33)
  concat_obj (Sea, Deck, Classes)
  concat_obj (Classes, Walls, Classes)
  concat_obj (Classes, Chimney, Classes)
  dev_set_color (Color[0])
  dev_display (Deck)
  dev_set_color (Color[1])
  dev_display (Sea)
  dev_set_color (Color[2])
  dev_display (Walls)
  dev_set_color (Color[3])
  dev_display (Chimney)
  Message := ‘Training regions for the color classifier’
  disp_message (WindowHandle, Message, ‘window’, 12, 12, ‘black’, ‘true’)
  disp_continue_message (WindowHandle, ‘black’, ‘true’)
  stop ()

• 創建gmm分類器

• 之前的博客已經介紹了使用方法https://blog.csdn.net/cashmood/article/details/90664731
  create_class_gmm (3, 4, [1,10], ‘full’, ‘none’, 2, 42, GMMHandle)

• 向GMM分類器中添加訓練樣本
  add_samples_image_class_gmm (Image, Classes, GMMHandle, 2.0)

• 顯示圖片
  dev_display (Image)

• 定義一個字符串變量，并進行初始化
  Message := ‘Training …’
  disp_message (WindowHandle, Message, ‘window’, 12, 12, ‘black’, ‘true’)

• 訓練分類器
  train_class_gmm (GMMHandle, 500, 1e-4, ‘uniform’, 1e-4, Centers, Iter)
  Message := Message + ’ ready.’
  Message[1] := ‘Segment image using the classifier …’
  disp_message (WindowHandle, Message, ‘window’, 12, 12, ‘black’, ‘true’)

• clear_samples_class_gmm - 清除高斯混合模型的訓練數據。主要是了釋放內存。
  clear_samples_class_gmm (GMMHandle)

• 使用GMM分類器對圖片區域進行分類
  classify_image_class_gmm (Image, ClassRegions, GMMHandle, 0.0001)

• 使用區域的平均灰度值繪制區域。
  region_to_mean (ClassRegions, Image, ImageClass)

• 顯示
  dev_display (ImageClass)
  Message[1] := Message[1] + ’ ready.’
  disp_message (WindowHandle, Message, ‘window’, 12, 12, ‘black’, ‘true’)

• 清除GMM分類器的句柄，釋放內存
  clear_class_gmm (GMMHandle)

處理思路

這個例子介紹了使用GMM分類器對彩色圖片進行分類的例子，選取多通道圖片的感興趣區域與背景可以對多通道圖片進行快速分類。

后記

大家有什么問題可以向我提問哈，我看到了第一時間回復，希望在學習的路上多多結交良師益友。