fr/main.py

42 lines
1.2 KiB
Python
Executable File

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