train

getval_imgs.py

@@ -0,0 +1,22 @@
+import os,shutil
+
+allimgs_path = "paper_data/images"
+valimgs_path = "paper_data/val_imgs"
+valtxt_path = "paper_data/ImageSets/Main/val.txt"
+
+with open(valtxt_path, "r", encoding='UTF-8') as val_file:
+    val_lines = val_file.readlines()
+lines_num = len(val_lines)
+print("lines_num:",lines_num)
+i=0
+for val_line in val_lines:
+
+    imgname = val_line.strip("\n").strip("\r")+".jpg"
+    print(imgname)
+    if imgname in os.listdir(allimgs_path):
+        img_path = f"{allimgs_path}/{imgname}"
+        shutil.copy(img_path,valimgs_path)
+        if i == lines_num:
+            break
+        i+=1
+