广州视觉芯软件有限公司

广州视觉芯软件有限公司

* 创建一个新的字典
create_dict(ObjectDictionary)

* 类别名称列表和对应的ID列表
class_names := ['Defect1','Defect2','Defect3','Defect4','Defect5','Defect6','Defect7','Defect8','Defect9','Defect10','Defect11','Defect12','Defect13','Defect14','Defect15','Defect16','Defect17','Defect18']
class_ids := [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17]
image_dir := 'c:/视觉芯/images/'
* 保存类别和ID到字典中
set_dict_tuple(ObjectDictionary, 'ClassIDs', class_ids)
set_dict_tuple(ObjectDictionary, 'ClassNames', class_names)
set_dict_tuple(ObjectDictionary, 'ImageDir', image_dir)

list_image_files ('images', 'default', 'recursive', ImageFiles)
tuple_shuffle (ImageFiles, DataList)

AllSamples:=[]
for Index := 0 to 9 by 1
    imageFile:=DataList[0]
    tuple_length (DataList, Length)
    create_dict (SampleImage)
    set_dict_tuple (SampleImage, 'image_id', Index+1)
    set_dict_tuple (SampleImage, 'image_file_name', imageFile)
    
    bbox_label_id:=[1]
    bbox_row1:=[1]
    bbox_col1:=[2]
    bbox_row2:=[3]
    bbox_col2:=[4]
    
 
    set_dict_tuple (SampleImage, 'bbox_label_id', bbox_label_id)
    set_dict_tuple (SampleImage, 'bbox_row1', bbox_row1)
    set_dict_tuple (SampleImage, 'bbox_col1', bbox_col1)
    set_dict_tuple (SampleImage, 'bbox_row2', bbox_row2)
    set_dict_tuple (SampleImage, 'bbox_col2', bbox_col2)
    
    tuple_concat (AllSamples, SampleImage, AllSamples)
endfor
set_dict_tuple(ObjectDictionary, 'samples', AllSamples)
* 保存字典到.hdict文件
write_dict (ObjectDictionary, 'dl_test.hdict', [], [])

03-29 06:01