import os import sys op_path = "dataset" if len(sys.argv) < 2: print("no input raw data folder") exit(0) rdat = sys.argv[1] print(rdat, [name for name in os.listdir(rdat) if os.path.isdir(os.path.join(rdat, name))]) subj_names = [name for name in os.listdir(rdat) if os.path.isdir(os.path.join(rdat, name)) ] os.system("mkdir -p dataset") for d in subj_names: inpath = rdat + "/" + d outpath = op_path + "/" + d os.system("mkdir -p " + outpath) img_files = [name for name in os.listdir(inpath) if not os.path.isdir(os.path.join(inpath, name)) ] # print(img_files) for img in img_files: inimg = inpath + "/" + img # print(inimg) # os.system("convert -verbose -coalesce " + inimg + " " + inimg + ".png") # os.system("rm "+ inimg) os.system("python3 face_cutter.py " + inimg + " " + outpath ) inpath2 = outpath outpath2 = outpath img_files2 = [name for name in os.listdir(inpath2) if not os.path.isdir(os.path.join(inpath2, name)) ] print(inpath2,img_files2) for img2 in img_files2: inimg2 = inpath2 + "/" + img2 os.system("python3 hogger.py " + inimg2 + " " + outpath2 ) os.system("rm " + inimg2) # break