From f15611421eab3c3614e1a83a65db4d9bde98cc2f Mon Sep 17 00:00:00 2001
From: "Eric.Lee" <xiang.li@ivglass.com>
Date: Mon, 16 Aug 2021 16:00:58 +0800
Subject: [PATCH] update

---
 utils/datasets_voc.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/utils/datasets_voc.py b/utils/datasets_voc.py
index b94bef2..4471f8c 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
-- 
GitLab