diff --git a/README.md b/README.md index 73c7b3dd6b663a1f22d7e20afab887932208c412..69eccea3db856f0d8a55c541007652ce9d471f27 100644 --- a/README.md +++ b/README.md @@ -67,10 +67,18 @@ The difference between this repo and GDR-Net (CVPR2021) mainly including: `./core/gdrn_modeling/train_gdrn.sh <config_path> <gpu_ids> (other args)` +For example: + +`./core/gdrn_modeling/train_gdrn.sh configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py 0` + ### Testing `./core/gdrn_modeling/test_gdrn.sh <config_path> <gpu_ids> <ckpt_path> (other args)` +For example: + +`./core/gdrn_modeling/test_gdrn.sh configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py 0 output/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv/model_final_wo_optim.pth` + ## Pose Refinement We utilize depth information to further refine the estimated pose. diff --git a/configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py b/configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py index 823c7ad7c018dd302f131de6b07784c69d55b2c1..b167a6c65f29fdd8eb8dc32f0485f15a9f41ec7b 100644 --- a/configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py +++ b/configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py @@ -1,7 +1,7 @@ # about 3 days _base_ = ["../../_base_/gdrn_base.py"] -OUTPUT_DIR = "output/gdrn/ycbv/convnext_a6_cPnP_AugCosyAAETrunc_BG05_Dmask_amodalClipBox_ycbv_real_pbr_visib30_10e" +OUTPUT_DIR = "output/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv" INPUT = dict( DZI_PAD_SCALE=1.5, TRUNCATE_FG=True,