自己写了个开源数据集转hdict的脚本

本来想直接论坛找现成的无奈找到的各种版本都有问题，所以自己写了个

halcon深度检测效果好吗？

xuyongze 发表于 2022-4-6 21:15
halcon深度检测效果好吗？

还在测试中

1. read_dict ('C:/Users/12820/Desktop/数据/分割.hdict', [], [], DictHandle)
2. * Image Acquisition 01: Code generated by Image Acquisition 01
4. create_dict (NEWDictHandle1)
5. class_ids:=[0,1,2,3,4,5]
6. class_names:=['孔','印刷','断裂','短','多','线条']
7. image_dir:='D:/BaiduNetdiskDownload/PCB_DATASET/PCB_DATASET/images/'
9. set_dict_tuple (NEWDictHandle1, 'class_ids', class_ids)
10. set_dict_tuple (NEWDictHandle1, 'class_names', class_names)
11. set_dict_tuple (NEWDictHandle1, 'image_dir', image_dir)
13. list_files ('C:/Users/12820/Desktop/分割测试文件夹/image', ['files','follow_links','recursive'], ImageFiles)
14. tuple_regexp_select (ImageFiles, ['\\.(tif|tiff|gif|bmp|jpg|jpeg|jp2|png|pcx|pgm|ppm|pbm|xwd|ima|hobj)$','ignore_case'], ImageFiles)
17. list_files ('C:/Users/12820/Desktop/分割测试文件夹/anno', ['files','follow_links','recursive'], xmladdress)
18. samples:=[]
19. for Index := 0 to |ImageFiles| - 1 by 1
20.     read_image (Image, ImageFiles[Index])
21.    
22.    
23.    
24.    
25.    
26.    open_file (xmladdress[Index], 'input', FileHandle)
27.    IsEof := false
28.    bbox_row1:=[]  
29.    bbox_col1:=[]  
30.    bbox_row2:=[]  
31.    bbox_col2:=[]  
32.    bbox_label_id:=[]
33.    while (not(IsEof))
34.        fread_line (FileHandle, XmlElement, IsEof)
35.        if (IsEof)
36.            break
37.        endif
38.        tuple_split (XmlElement, '<''>', Substrings)
39.        create_dict (image)
40.        if (Substrings[1]=='folder')
41.            floder:= Substrings[2]
42.        endif  
43.        if (Substrings[1]=='filename')
44.            filename:= Substrings[2]
45.        endif   
46.        if (Substrings[1]=='name')
47.            if (Substrings[2]=='missing_hole' )
48.                bbox_label_id:=[bbox_label_id,0]
49.            elseif (Substrings[2]=='mouse_bite')
50.                bbox_label_id:=[bbox_label_id,1]
51.            elseif (Substrings[2]=='open_circuit')
52.                bbox_label_id:=[bbox_label_id,2]
53.            elseif (Substrings[2]=='short')
54.                     bbox_label_id:=[bbox_label_id,3]
55.            elseif (Substrings[2]=='spur')
56.                bbox_label_id:=[bbox_label_id,4]
57.            elseif (Substrings[2]=='spurious_copper')
58.                bbox_label_id:=[bbox_label_id,5]
59.            endif
60.       endif
61.       
62.       if (Substrings[1]=='xmin')
63.           bbox_col1:= [bbox_col1,Substrings[2]]
64.           tuple_number (bbox_col1, bbox_col1)
65.    
66.       endif  
67.       if (Substrings[1]=='ymin')
68.           bbox_row1:= [bbox_row1,Substrings[2] ]
69.           tuple_number (bbox_row1, bbox_row1)
70.       endif  
71.       if (Substrings[1]=='xmax')
72.           bbox_col2:=[bbox_col2, Substrings[2] ]
73.           tuple_number (bbox_col2, bbox_col2)
74.       endif  
75.       if (Substrings[1]=='ymax')
76.           bbox_row2:=[bbox_row2, Substrings[2] ]
77.           tuple_number (bbox_row2, bbox_row2)
78.       endif  
81.     endwhile
82.    
83.    
84.    * gen_rectangle1 (Rectangle,bbox_row1 , bbox_col1,bbox_row2 , bbox_col2)
85.    
86.   set_dict_tuple (image, 'image_id', Index+1)
87.   set_dict_tuple (image, 'image_file_name', floder+'/'+filename)
88.   set_dict_tuple (image, 'bbox_label_id', bbox_label_id)
89.   set_dict_tuple (image, 'bbox_row1', bbox_row1)
90.   set_dict_tuple (image, 'bbox_col1', bbox_col1)
91.   set_dict_tuple (image, 'bbox_row2', bbox_row2)
92.   set_dict_tuple (image, 'bbox_col2', bbox_col2)
93. samples:=[samples,image]
94.    
95.     *stop()  
96. endfor
98. set_dict_tuple (NEWDictHandle1, 'samples', samples)
100. write_dict (NEWDictHandle1, 'C:/Users/12820/Desktop/数据/全电路板数据.hdict', [], [])

复制代码