From bae23a6bc4b3940e7685e1a5806b206fa756e681 Mon Sep 17 00:00:00 2001
From: liuxingyu <lxy17@foxmail.com>
Date: Wed, 2 Nov 2022 13:41:39 +0800
Subject: [PATCH] upd example for training/testing gdrn
---
README.md | 8 ++++++++
...EGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py | 2 +-
2 files changed, 9 insertions(+), 1 deletion(-)
diff --git a/README.md b/README.md
index 73c7b3d..69eccea 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 823c7ad..b167a6c 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,
--
GitLab