23 lines
596 B
Python
23 lines
596 B
Python
import os,shutil
|
|
|
|
allimgs_path = "paper_data/images"
|
|
valimgs_path = "paper_data/val_imgs_7.1"
|
|
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
|
|
|