add cfg for tless

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/1.py

+_base_ = ["../../../_base_/gdrn_base.py"]
+
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/1"
+INPUT = dict(
+    DZI_PAD_SCALE=1.5,
+    TRUNCATE_FG=True,
+    CHANGE_BG_PROB=0.5,
+    COLOR_AUG_PROB=0.8,
+    MIN_SIZE_TRAIN=540,
+    MAX_SIZE_TRAIN=720,
+    MIN_SIZE_TEST=540,
+    MAX_SIZE_TEST=720,
+    IMG_AUG_RESIZE=False,
+    COLOR_AUG_TYPE="code",
+    COLOR_AUG_CODE=(
+        "Sequential(["
+        # Sometimes(0.5, PerspectiveTransform(0.05)),
+        # Sometimes(0.5, CropAndPad(percent=(-0.05, 0.1))),
+        # Sometimes(0.5, Affine(scale=(1.0, 1.2))),
+        "Sometimes(0.5, CoarseDropout( p=0.2, size_percent=0.05) ),"
+        "Sometimes(0.4, GaussianBlur((0., 3.))),"
+        "Sometimes(0.3, pillike.EnhanceSharpness(factor=(0., 50.))),"
+        "Sometimes(0.3, pillike.EnhanceContrast(factor=(0.2, 50.))),"
+        "Sometimes(0.5, pillike.EnhanceBrightness(factor=(0.1, 6.))),"
+        "Sometimes(0.3, pillike.EnhanceColor(factor=(0., 20.))),"
+        "Sometimes(0.5, Add((-25, 25), per_channel=0.3)),"
+        "Sometimes(0.3, Invert(0.2, per_channel=True)),"
+        "Sometimes(0.5, Multiply((0.6, 1.4), per_channel=0.5)),"
+        "Sometimes(0.5, Multiply((0.6, 1.4))),"
+        "Sometimes(0.1, AdditiveGaussianNoise(scale=10, per_channel=True)),"
+        "Sometimes(0.5, iaa.contrast.LinearContrast((0.5, 2.2), per_channel=0.3)),"
+        "Sometimes(0.5, Grayscale(alpha=(0.0, 1.0))),"  # maybe remove for det
+        "], random_order=True)"
+        # cosy+aae
+    ),
+)
+
+SOLVER = dict(
+    IMS_PER_BATCH=36,
+    TOTAL_EPOCHS=100,
+    LR_SCHEDULER_NAME="flat_and_anneal",
+    ANNEAL_METHOD="cosine",  # "cosine"
+    ANNEAL_POINT=0.72,
+    OPTIMIZER_CFG=dict(_delete_=True, type="Ranger", lr=8e-4, weight_decay=0.01),
+    WEIGHT_DECAY=0.0,
+    WARMUP_FACTOR=0.001,
+    WARMUP_ITERS=1000,
+)
+
+DATASETS = dict(
+    TRAIN=("tless_1_train_pbr",),
+    TRAIN2=("tless_1_train_primesense"),
+    TRAIN2_RATIO=0.3,
+    TEST=("tless_bop_test_primesense",),
+    # AP        AP50    AP75    AR      inf.time
+    DET_FILES_TEST=("datasets/BOP_DATASETS/tless/test/test_bboxes/yolox_x_640_tless_real_pbr_tless_bop_test.json",),
+    DET_TOPK_PER_OBJ=100,
+)
+
+DATALOADER = dict(
+    # Number of data loading threads
+    NUM_WORKERS=8,
+    FILTER_VISIB_THR=0.3,
+)
+
+MODEL = dict(
+    LOAD_DETS_TEST=True,
+    PIXEL_MEAN=[0.0, 0.0, 0.0],
+    PIXEL_STD=[255.0, 255.0, 255.0],
+    BBOX_TYPE="AMODAL_CLIP",  # VISIB or AMODAL
+    POSE_NET=dict(
+        NAME="GDRN_double_mask",
+        XYZ_ONLINE=True,
+        BACKBONE=dict(
+            FREEZE=False,
+            PRETRAINED="timm",
+            INIT_CFG=dict(
+                type="timm/convnext_base",
+                pretrained=True,
+                in_chans=3,
+                features_only=True,
+                out_indices=(3,),
+            ),
+        ),
+        ## geo head: Mask, XYZ, Region
+        GEO_HEAD=dict(
+            FREEZE=False,
+            INIT_CFG=dict(
+                type="TopDownDoubleMaskXyzRegionHead",
+                in_dim=1024,  # this is num out channels of backbone conv feature
+            ),
+            NUM_REGIONS=64,
+        ),
+        PNP_NET=dict(
+            INIT_CFG=dict(norm="GN", act="gelu"),
+            REGION_ATTENTION=True,
+            WITH_2D_COORD=True,
+            ROT_TYPE="allo_rot6d",
+            TRANS_TYPE="centroid_z",
+        ),
+        LOSS_CFG=dict(
+            # xyz loss ----------------------------
+            XYZ_LOSS_TYPE="L1",  # L1 | CE_coor
+            XYZ_LOSS_MASK_GT="visib",  # trunc | visib | obj
+            XYZ_LW=1.0,
+            # mask loss ---------------------------
+            MASK_LOSS_TYPE="L1",  # L1 | BCE | CE
+            MASK_LOSS_GT="trunc",  # trunc | visib | gt
+            MASK_LW=1.0,
+            # full mask loss ---------------------------
+            FULL_MASK_LOSS_TYPE="L1",  # L1 | BCE | CE
+            FULL_MASK_LW=1.0,
+            # region loss -------------------------
+            REGION_LOSS_TYPE="CE",  # CE
+            REGION_LOSS_MASK_GT="visib",  # trunc | visib | obj
+            REGION_LW=1.0,
+            # pm loss --------------
+            PM_LOSS_SYM=True,  # NOTE: sym loss
+            PM_R_ONLY=True,  # only do R loss in PM
+            PM_LW=1.0,
+            # centroid loss -------
+            CENTROID_LOSS_TYPE="L1",
+            CENTROID_LW=1.0,
+            # z loss -----------
+            Z_LOSS_TYPE="L1",
+            Z_LW=1.0,
+        ),
+    ),
+)
+
+TEST = dict(EVAL_PERIOD=0, VIS=False, TEST_BBOX_TYPE="est")  # gt | est
+
+VAL = dict(
+    DATASET_NAME="tless",
+    SCRIPT_PATH="lib/pysixd/scripts/eval_pose_results_more.py",
+    TARGETS_FILENAME="test_targets_bop19.json",
+    ERROR_TYPES="mspd,mssd,vsd,ad,reS,teS",
+    RENDERER_TYPE="cpp",  # cpp, python, egl
+    SPLIT="test",
+    SPLIT_TYPE="",
+    N_TOP=-1,  # SISO: 1, VIVO: -1 (for LINEMOD, 1/-1 are the same)
+    EVAL_CACHED=False,  # if the predicted poses have been saved
+    SCORE_ONLY=False,  # if the errors have been calculated
+    EVAL_PRINT_ONLY=False,  # if the scores/recalls have been saved
+    EVAL_PRECISION=False,  # use precision or recall
+    USE_BOP=True,  # whether to use bop toolkit
+)

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/10.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/10"
+DATASETS = dict(TRAIN=("tless_10_train_pbr",), TRAIN2=("tless_10_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/11.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/11"
+DATASETS = dict(TRAIN=("tless_11_train_pbr",), TRAIN2=("tless_11_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/12.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/12"
+DATASETS = dict(TRAIN=("tless_12_train_pbr",), TRAIN2=("tless_12_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/13.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/13"
+DATASETS = dict(TRAIN=("tless_13_train_pbr",), TRAIN2=("tless_13_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/14.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/14"
+DATASETS = dict(TRAIN=("tless_14_train_pbr",), TRAIN2=("tless_14_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/15.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/15"
+DATASETS = dict(TRAIN=("tless_15_train_pbr",), TRAIN2=("tless_15_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/16.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/16"
+DATASETS = dict(TRAIN=("tless_16_train_pbr",), TRAIN2=("tless_16_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/17.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/17"
+DATASETS = dict(TRAIN=("tless_17_train_pbr",), TRAIN2=("tless_17_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/18.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/18"
+DATASETS = dict(TRAIN=("tless_18_train_pbr",), TRAIN2=("tless_18_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/19.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/19"
+DATASETS = dict(TRAIN=("tless_19_train_pbr",), TRAIN2=("tless_19_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/2.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/2"
+DATASETS = dict(TRAIN=("tless_2_train_pbr",), TRAIN2=("tless_2_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/20.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/20"
+DATASETS = dict(TRAIN=("tless_20_train_pbr",), TRAIN2=("tless_20_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/21.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/21"
+DATASETS = dict(TRAIN=("tless_21_train_pbr",), TRAIN2=("tless_21_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/22.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/22"
+DATASETS = dict(TRAIN=("tless_22_train_pbr",), TRAIN2=("tless_22_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/23.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/23"
+DATASETS = dict(TRAIN=("tless_23_train_pbr",), TRAIN2=("tless_23_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/24.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/24"
+DATASETS = dict(TRAIN=("tless_24_train_pbr",), TRAIN2=("tless_24_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/25.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/25"
+DATASETS = dict(TRAIN=("tless_25_train_pbr",), TRAIN2=("tless_25_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/26.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/26"
+DATASETS = dict(TRAIN=("tless_26_train_pbr",), TRAIN2=("tless_26_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/27.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/27"
+DATASETS = dict(TRAIN=("tless_27_train_pbr",), TRAIN2=("tless_27_train_primesense",))