Update remove_img_by_list.py

marxmarma · Aug 13, 2020 · c64637d · c64637d
1 parent 9819dd6
commit c64637d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/remove_img_by_list.py b/remove_img_by_list.py
@@ -11,6 +11,7 @@
 import shutil
 import sys
 import argparse
+from tqdm import tqdm
 
 
 def parse_args():
@@ -38,18 +39,15 @@ def pick(delListTxt, imgDir):
     os.makedirs(pickedImg_dir)
 
     file_ = open(delListTxt, 'r')
-    delimglists = file_.readlines()
+    delimglists = [f.strip() for f in file_.readlines()]
     file_.close()
 
-    for img in imglists:
+    for img in tqdm(imglists):
         img = img.strip()
         imgname = os.path.basename(img)
-        if img in delimglists:
+        if imgname in delimglists:
             continue
 
-        print(imgname + " -->start!")
-        fileInfor = imgname.split(".")
-
         src_img = os.path.join(imgDir, img)
         dst_img = os.path.join(pickedImg_dir, imgname)