Skip to content
Snippets Groups Projects
Commit bae23a6b authored by liuxingyu's avatar liuxingyu
Browse files

upd example for training/testing gdrn

parent 86b6dd4b
No related branches found
No related tags found
No related merge requests found
......@@ -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.
......
# 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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment