copy_val_for_detect.py 921 B

12345678910111213141516171819202122232425262728
  1. import os.path
  2. import shutil
  3. file="D:/data/data_train_val_test/val.txt"
  4. dst_path="D:/data/val"
  5. if os.path.exists(dst_path):
  6. shutil.rmtree(dst_path)
  7. os.mkdir(dst_path)
  8. with open(file,"r",encoding="utf-8") as f:
  9. i = 0
  10. j = 0
  11. lines = f.readlines()
  12. for line in lines:
  13. line= line.strip()
  14. src_img_bsname=os.path.basename(line)
  15. src_img_dirname = os.path.basename(os.path.dirname(line))
  16. #print("src_img_dirname: ", src_img_dirname)
  17. src_json_bsname = src_img_bsname.rsplit(".",1)[0]+".json"
  18. shutil.copy(line,os.path.join(dst_path,src_img_dirname+"+"+src_img_bsname))
  19. src_json_path = os.path.dirname(line)+"/"+src_json_bsname
  20. print(src_json_path)
  21. j +=1
  22. if os.path.exists(src_json_path):
  23. shutil.copy(src_json_path,os.path.join(dst_path,src_img_dirname+"+"+src_json_bsname))
  24. i += 1
  25. print("j ,i: ", j, i)