copy_eval_for_detect.py 894 B

1234567891011121314151617181920212223242526
  1. import os.path
  2. import shutil
  3. file= "eval.txt"
  4. dst_path="D:/data/eval"
  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)