IF3D / preprocessing /run_preprocessing.sh
leobcc's picture
vid2avatar baseline
6325697
raw
history blame
1.19 kB
# pre-define data
source="custom"
seq=""
gender="FEMALE"
# run ROMP to get initial SMPL parameters
echo "Running ROMP"
romp --mode=video --calc_smpl --render_mesh -i=./raw_data/$seq/frames -o=./raw_data/$seq/ROMP --smpl_path ./romp/SMPL_$gender.pth
# obtain the projected masks through estimated perspective camera (so that OpenPose detection)
echo "Getting projected SMPL masks"
python preprocessing.py --source $source --seq $seq --gender $gender --mode mask
# run OpenPose to get 2D keypoints
echo "Running OpenPose"
python run_openpose.py --openpose_dir {PATH_TO_OPENPOSE} --seq $seq
# offline refine poses
echo "Refining poses offline"
python preprocessing.py --source $source --seq $seq --gender $gender --mode refine
# scale images and center the human in 3D space
echo "Scaling images and centering human in 3D space"
python preprocessing.py --source $source --seq $seq --gender $gender --mode final --scale_factor 1
# normalize cameras such that all cameras are within the sphere of radius 3
echo "Normalizing cameras"
python normalize_cameras.py --input_cameras_file ../data/$seq/cameras.npz \
--output_cameras_file ../data/$seq/cameras_normalize.npz