Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import glob, os, sys
- from shutil import copyfile
- args = sys.argv
- path = "."
- dir_path = "Valid"
- if len(args) >= 2:
- path = args[1]
- if len(args) >= 3:
- dir_path = args[2]
- print("PATH:", path, sep="\t")
- os.chdir(path)
- imgs = glob.glob("*.jpg")
- xmls = glob.glob("*.xml")
- name_sep = lambda x: x.split(".")[0]
- i = map(name_sep, imgs)
- x = map(name_sep, xmls)
- diff = set(i)^set(x)
- print(".jpg:\t", len(imgs),"\n.xml:\t", len(xmls), sep="")
- print(".imgs > .xmls" if max(imgs, xmls, key=lambda x: len(x)) else ".xmls > .imgs")
- print("Difference:",len(diff))
- for file in diff:
- print("\t\t",file,sep="")
- i = map(name_sep, imgs)
- x = map(name_sep, xmls)
- intersect = set(i)&set(x)
- c = 0
- print("Pre-valid pairs, total: ", len(intersect), sep="\t")
- try:
- print("DIR:", dir_path, sep="\t", end="\t")
- os.mkdir(dir_path)
- print("CREATED")
- for file in intersect:
- i_file = file+".jpg"
- x_file = file+".xml"
- #print(i_file, x_file, dir_path)
- copyfile(i_file, dir_path+"\\"+i_file)
- copyfile(x_file, dir_path+"\\"+x_file)
- c += 1
- except FileExistsError:
- print("ERROR. Dir already exists")
- except Exception as err:
- print("Emmm...", err)
- print("COPY, total:", c, sep="\t")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement