Halcon自动化训练OCR分类器举例


 

进行OCR的时候,同一个字符,通常需要很多字符小图,这样图片多了,才能训练出健壮的分类器。如何自动化地执行该过程呢?我提供一种思路。

 

待训练的图片集如下:

其中,“文件夹0”和“文件夹1”中有多张图片,例如“文件夹1”中的图片是这样的(图片名称第一个字符都是“1”):

 

训练OCR分类器的完整代码如下:

 1 set_font (200000, 'Arial'+'-Bold-20')  2 dev_set_draw ('margin')  3 
 4 ImageFolder := '第2行data'
 5 
 6 
 7 list_files (ImageFolder, ['files','follow_links','recursive'], ImageFiles)  8 tuple_regexp_select (ImageFiles, ['\\.(tif|tiff|gif|bmp|jpg|jpeg|jp2|png|pcx|pgm|ppm|pbm|xwd|ima)$','ignore_case'], ImageFiles)  9 
10 gen_empty_obj (Images) 11 gen_empty_obj (RegionChars) 12 Chars := [] 13 
14 for Index := 0 to |ImageFiles| - 1 by 1
15  read_image (Image, ImageFiles[Index]) 16  parse_filename (ImageFiles[Index], BaseName, Extension, Directory) 17  count_channels (Image, Channels) 18     if (Channels != 1) 19  rgb1_to_gray (Image, Image) 20  endif 21     
22     *取文件名的第一个字符(例如字符1可能有多张图片,图片原始名字可能为11.bmp、12.bmp、13.bmp……) 23     tuple_substr (BaseName, 0, 0, BaseNameReal) 24     
25     *增强图像对比度 26  scale_image_max (Image, Image) 27     emphasize (Image, Image, 11, 11, 1) 28 
29     if (BaseNameReal == '.') 30         BaseNameReal := '.'
31  endif 32     
33     threshold (Image, Region, 0, 152) 34 *     binary_threshold (Image, Region, 'max_separability', 'dark', UsedThreshold) 35     opening_circle (Region, Region, 1) 36     fill_up_shape (Region, Region, 'area', 1, 100) 37     
38     *将字符图像、字符区域、字符文本分别添加到Images、RegionChars、Chars中 39  concat_obj (Images, Image, Images) 40  concat_obj (RegionChars, Region, RegionChars) 41     Chars := [Chars,BaseNameReal]  42     
43  dev_display (Image) 44  dev_display (Region) 45 endfor 46 
47 *1.1创建训练文件 48 TrainFile:='Line2_Words.trf'
49 * delete_file(TrainFile) 50 
51 *1.2将字符区域、字符图像与字符文本关联,保存到.trf训练文件中 52 for i:=1 to |Chars| by 1
53  select_obj(RegionChars, SingleWord, i) 54  select_obj(Images, SingleImage, i) 55     append_ocr_trainf(SingleWord,SingleImage,Chars[i-1],TrainFile) 56 endfor 57 
58 
59 ***********************************************************************
60 
61 *2.1确定字体分类器文件名 62 FontFile:='Line2_Words.omc'
63 
64 *2.2得到字符标识名(这一步非必须) 65 read_ocr_trainf_names(TrainFile, CharacterNames, CharacterCount) 66 
67 *2.3确定神经网络隐藏层节点数 68 NumHidden := 80
69 
70 *2.4创建神经网络分类器 71 create_ocr_class_mlp(42, 67, 'constant', 'default', CharacterNames, NumHidden, 'none', 10, 42, OCRHandle) 72 
73 *2.5训练神经网络 74 trainf_ocr_class_mlp(OCRHandle, TrainFile, 200, 1, 0.9, Error, ErrorLog) 75 
76 *2.6保存训练结果 77 write_ocr_class_mlp(OCRHandle, FontFile) 78 
79 *2.7清除句柄 80 clear_ocr_class_mlp(OCRHandle)

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM