提交 f1561142 编写于 作者: E Eric.Lee

update

上级 6c4b51d5
......@@ -59,6 +59,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
print("label_voc_dict : {}".format(label_voc_dict))
img_files = []
label_files = []
s_idx = 0
for file in os.listdir(path):
if ".jpg" in file:
path_img = path + file
......@@ -71,6 +72,9 @@ class LoadImagesAndLabels(Dataset): # for training/testing
obj_num_sum += obj_num
img_files.append(path_img)
label_files.append(path_label)
s_idx += 1
print(" Init LoadImagesAndLabels <{:6d}> - images ".format(s_idx),end = "\r")
print()
self.label_voc_dict = label_voc_dict
self.img_files = img_files
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册