12345678910111213141516171819202122232425262728 |
- import os.path
- import shutil
- file="D:/data/data_train_val_test/val.txt"
- dst_path="D:/data/val"
- if os.path.exists(dst_path):
- shutil.rmtree(dst_path)
- os.mkdir(dst_path)
- with open(file,"r",encoding="utf-8") as f:
- i = 0
- j = 0
- lines = f.readlines()
- for line in lines:
- line= line.strip()
- src_img_bsname=os.path.basename(line)
- src_img_dirname = os.path.basename(os.path.dirname(line))
- #print("src_img_dirname: ", src_img_dirname)
- src_json_bsname = src_img_bsname.rsplit(".",1)[0]+".json"
- shutil.copy(line,os.path.join(dst_path,src_img_dirname+"+"+src_img_bsname))
- src_json_path = os.path.dirname(line)+"/"+src_json_bsname
- print(src_json_path)
- j +=1
- if os.path.exists(src_json_path):
- shutil.copy(src_json_path,os.path.join(dst_path,src_img_dirname+"+"+src_json_bsname))
- i += 1
- print("j ,i: ", j, i)
|