File size: 1,194 Bytes
6325697 |
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 |
# 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 |