forked from 170010011/fr
42 lines
1.2 KiB
Python
Executable File
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
|
|
|