Skip to content

Commit

Permalink
update readme and eval scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
lixiny committed Feb 15, 2022
1 parent 502a53c commit a4867bc
Show file tree
Hide file tree
Showing 9 changed files with 227 additions and 270 deletions.
382 changes: 112 additions & 270 deletions README.md

Large diffs are not rendered by default.

19 changes: 19 additions & 0 deletions scripts/dump_HoNetPiCR_FHB.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
echo "Using GPU_ID: ${GPU_ID}"

python ./scripts/dump_picr_res.py \
--gpu ${GPU_ID} \
--dist_master_addr localhost \
--dist_master_port 12355 \
--exp_keyword fhb \
--train_datasets fhb \
--train_splits train \
--val_dataset fhb \
--val_split test \
--split_mode actions \
--batch_size 8 \
--dump_eval \
--dump \
--vertex_contact_thresh 0.8 \
--filter_thresh 5.0 \
--dump_prefix common/picr \
--init_ckpt CPF_checkpoints/picr/fhb/checkpoint_200.pth.tar
19 changes: 19 additions & 0 deletions scripts/dump_HoNetPiCR_HO3Dv1.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/dump_picr_res.py \
--gpu ${GPU_ID} \
--dist_master_addr localhost \
--dist_master_port 12356 \
--exp_keyword ho3dv1 \
--train_datasets ho3d \
--train_splits train \
--val_dataset ho3d \
--val_split test \
--split_mode objects \
--batch_size 4 \
--dump_eval \
--dump \
--vertex_contact_thresh 0.8 \
--filter_thresh 5.0 \
--dump_prefix common/picr_ho3dv1 \
--init_ckpt CPF_checkpoints/picr/ho3dv1/checkpoint_300.pth.tar
20 changes: 20 additions & 0 deletions scripts/dump_HoNetPiCR_HO3Dv2.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/dump_picr_res.py \
--gpu ${GPU_ID} \
--dist_master_addr localhost \
--dist_master_port 12356 \
--exp_keyword ho3dofficial \
--train_datasets ho3d \
--train_splits val \
--val_dataset ho3d \
--val_split test \
--split_mode official \
--batch_size 4 \
--dump_eval \
--dump \
--test_dump \
--vertex_contact_thresh 0.8 \
--filter_thresh 5.0 \
--dump_prefix common/picr_ho3dofficial \
--init_ckpt CPF_checkpoints/picr/ho3dofficial/checkpoint_300.pth.tar
8 changes: 8 additions & 0 deletions scripts/fit_GeO_handobj_FHB.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/fit_geo.py \
--gpu ${GPU_ID} \
--n_workers 16 \
--data_path common/picr/fhbhands/test_actions_mf1.0_rf0.25_fct5.0_ec \
--mode hand_obj \
--compensate_tsl
14 changes: 14 additions & 0 deletions scripts/fit_GeO_handobj_HO3Dv1.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/fit_geo.py \
--gpu ${GPU_ID} \
--n_workers 24 \
--data_path common/picr_ho3dv1/HO3D/test_objects_mf1_likev1_fct5.0_ec/ \
--lr 1e-2 \
--n_iter 500 \
--hodata_no_use_cache \
--lambda_contact_loss 10.0 \
--lambda_repulsion_loss 6.0 \
--repulsion_query 0.030 \
--repulsion_threshold 0.080 \
--mode hand_obj
14 changes: 14 additions & 0 deletions scripts/fit_GeO_handobj_HO3Dv2.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/fit_geo.py \
--gpu ${GPU_ID} \
--n_workers 24 \
--data_path common/picr_ho3dofficial/HO3D/test_official_mf1_likev1_fct\(x\)_ec/ \
--lr 1e-2 \
--n_iter 500 \
--hodata_no_use_cache \
--lambda_contact_loss 10.0 \
--lambda_repulsion_loss 2.0 \
--repulsion_query 0.030 \
--repulsion_threshold 0.080 \
--mode hand_obj
7 changes: 7 additions & 0 deletions scripts/fit_GeO_handonly_FHB.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/fit_geo.py \
--gpu ${GPU_ID} \
--n_workers 16 \
--data_path common/picr/fhbhands/test_actions_mf1.0_rf0.25_fct5.0_ec \
--mode hand
14 changes: 14 additions & 0 deletions scripts/fit_GeO_handonly_HO3Dv1.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
echo "Using GPU_ID: ${GPU_ID}"

python scripts/fit_geo.py \
--gpu ${GPU_ID} \
--n_workers 24 \
--data_path common/picr_ho3dv1/HO3D/test_objects_mf1_likev1_fct5.0_ec/ \
--lr 1e-2 \
--n_iter 500 \
--hodata_no_use_cache \
--lambda_contact_loss 10.0 \
--lambda_repulsion_loss 4.0 \
--repulsion_query 0.030 \
--repulsion_threshold 0.080 \
--mode hand

0 comments on commit a4867bc

Please sign in to comment.