diff --git a/utils/datasets_voc.py b/utils/datasets_voc.py index b94bef2425ff3a2cb99567f49d7b578de07d0d67..4471f8c9ab2de7e114fef04520851f55dde07b06 100644 --- a/utils/datasets_voc.py +++ b/utils/datasets_voc.py @@ -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