Advertisement
GKirill

Tensor_Pre-Validate_By_JPG-XML_Pairs

Dec 16th, 2019
345
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.27 KB | None | 0 0
  1. import glob, os, sys
  2. from shutil import copyfile
  3. args = sys.argv
  4. path = "."
  5. dir_path = "Valid"
  6.  
  7. if len(args) >= 2:
  8.     path = args[1]
  9. if len(args) >= 3:
  10.     dir_path = args[2]
  11.  
  12. print("PATH:", path, sep="\t")
  13. os.chdir(path)
  14. imgs = glob.glob("*.jpg")
  15. xmls = glob.glob("*.xml")
  16.  
  17. name_sep = lambda x: x.split(".")[0]
  18. i = map(name_sep, imgs)
  19. x = map(name_sep, xmls)
  20. diff = set(i)^set(x)
  21.  
  22. print(".jpg:\t", len(imgs),"\n.xml:\t", len(xmls), sep="")
  23. print(".imgs > .xmls" if max(imgs, xmls, key=lambda x: len(x)) else ".xmls > .imgs")
  24.  
  25. print("Difference:",len(diff))
  26. for file in diff:
  27.     print("\t\t",file,sep="")
  28.  
  29. i = map(name_sep, imgs)
  30. x = map(name_sep, xmls)
  31. intersect = set(i)&set(x)
  32.  
  33. c = 0
  34. print("Pre-valid pairs, total: ", len(intersect), sep="\t")
  35. try:
  36.     print("DIR:", dir_path, sep="\t", end="\t")
  37.     os.mkdir(dir_path)
  38.     print("CREATED")
  39.    
  40.     for file in intersect:
  41.         i_file = file+".jpg"
  42.         x_file = file+".xml"
  43.        
  44.         #print(i_file, x_file, dir_path)
  45.         copyfile(i_file, dir_path+"\\"+i_file)
  46.         copyfile(x_file, dir_path+"\\"+x_file)
  47.         c += 1
  48. except FileExistsError:
  49.     print("ERROR. Dir already exists")
  50. except Exception as err:
  51.     print("Emmm...", err)
  52.    
  53. print("COPY, total:", c, sep="\t")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement