赞
踩
* 创建一个新的字典
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', [], [])
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。