-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_people_snapshot.sh
63 lines (60 loc) · 3.74 KB
/
run_people_snapshot.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#bash run_people_snapshot.sh
seq=male-3-casual #female-3-casual
dataset_name=people_snapshot #people_snapshot
view=/
dataset_type=$(echo $dataset_name | awk -F"_" '{print $1}')
gender=$(echo $seq | awk -F"-" '{print $1}') #if you run the zju_mocap, please set it to be the correct gender
decoupling=True
if [ -z "$(ls -A "data/${dataset_name}/${seq}/images")" ]
then
echo "please download ${dataset_name} dataset. if you run the people_snapshot, please use instantavatar to generate the images"
exit 0
fi
#===========run the Self-Correction-Human-Parsing-master======
if [ -z "$(ls -A "data/${dataset_name}/${seq}/schp")" ]
then
echo "doing schp(only for people_snapshot dattaset)"
#mkdir -p data/${dataset_name}/${seq}/schp
python read_image_name.py --folder_path data/${dataset_name}/${seq}/images --output_path data/${dataset_name}/${seq}/images.txt
cd submodules/Self-Correction-Human-Parsing-master
echo "warning!!! go to make global_local_evaluate.py saving pth useless"
python mhp_extension/global_local_parsing/global_local_evaluate.py --data-dir ../../data/${dataset_name}/${seq}/ --split-name images --model-restore mhp_extension/pretrain_model/exp_schp_multi_cihp_global.pth --log-dir ../../data/${dataset_name}/${seq}/ --save-results
#python simple_extractor.py --input-dir ../../data/${dataset_name}/${seq}/images --output-dir ../../data/${dataset_name}/${seq}/schp #--model-restore ./pretrain_model/resnet101-imagenet.pth
#mv
cd ../
cd ../
mv data/${dataset_name}/${seq}/images_parsing data/${dataset_name}/${seq}/schp
fi
#===========run the frankmocap======================
if [ -z "$(ls -A "data/${dataset_name}/${seq}/mocap_input/mocap")" ]
then
echo "doing mocap"
mkdir -p data/${dataset_name}/${seq}/mocap_input
cd submodules/frankmocap-main
timeout 30 python -m demo.demo_bodymocap --input_path ../../data/${dataset_name}/${seq}/images${view} --out_dir ../../data/${dataset_name}/${seq}/mocap_input --save_pred_pkl
cd ../
cd ../
fi
#==========GT segmentation==========================
if [ -z "$(ls -A "data/${dataset_name}/${seq}/cloth_segmentation/tee/")" ]
then
echo "doing cloth segmentation"
mkdir -p "data/${dataset_name}/${seq}/cloth_segmentation"
python seg_image.py --task_root data/${dataset_name}/${seq} --mask_root schp${view} --input_name images${view}
fi
#=========cloth initalization========================
if [ -z "$(ls -A "data/${dataset_name}/${seq}/cloth_model")" ]
then
echo "doing cloth initalization and it may spend a long time"
#cd lib_isp
#python fitting_image.py --save_root data/${dataset_name}/${seq} --mask_root schp${view}
#cd ..
#mv lib_isp/skin_weights.pkl "data/${dataset_name}/${seq}/cloth_model/"
#mkdir -p "data/${dataset_name}/${seq}/cloth_model"
python cloth_init.py --cloth_root data/${dataset_name}/${seq}/cloth_model/ --gender $gender --decoupling $decoupling
#python cloth_init.py --cloth_root data/people_snapshot/female-4-casual/cloth_model/ --gender female --decoupling false
fi
#===========main function======================================
python solver.py --profile ./profiles/${dataset_type}/${dataset_type}_default.yaml --dataset $dataset_name --seq $seq --logbase ${dataset_type}_default --fast --no_eval --decoupling $decoupling --garment_init
# python solver.py --profile ./profiles/people/people_default.yaml --dataset people_snapshot --seq female-3-casual --logbase people_default --fast --no_eval --garment_init --decoupling True --combine_only people_default/female-4-casual;people_default/female-4-casual
# python solver.py --profile ./profiles/people/people_default.yaml --dataset people_snapshot --seq female-4-casual --logbase people_default --fast --no_eval --garment_init --decoupling True --discolor_only black;dark_green