add cfg for tless

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/28.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/28"
+DATASETS = dict(TRAIN=("tless_28_train_pbr",), TRAIN2=("tless_28_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/29.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/29"
+DATASETS = dict(TRAIN=("tless_29_train_pbr",), TRAIN2=("tless_29_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/3.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/3"
+DATASETS = dict(TRAIN=("tless_3_train_pbr",), TRAIN2=("tless_3_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/30.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/30"
+DATASETS = dict(TRAIN=("tless_30_train_pbr",), TRAIN2=("tless_30_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/4.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/4"
+DATASETS = dict(TRAIN=("tless_4_train_pbr",), TRAIN2=("tless_4_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/5.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/5"
+DATASETS = dict(TRAIN=("tless_5_train_pbr",), TRAIN2=("tless_5_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/6.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/6"
+DATASETS = dict(TRAIN=("tless_6_train_pbr",), TRAIN2=("tless_6_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/7.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/7"
+DATASETS = dict(TRAIN=("tless_7_train_pbr",), TRAIN2=("tless_7_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/8.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/8"
+DATASETS = dict(TRAIN=("tless_8_train_pbr",), TRAIN2=("tless_8_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/9.py

+_base_ = "./1.py"
+OUTPUT_DIR = "output/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/9"
+DATASETS = dict(TRAIN=("tless_9_train_pbr",), TRAIN2=("tless_9_train_primesense",))

configs/gdrn/tlessSO/convnext_a6_cPnP_AugCosyAAE_BG05_Dmask_amodalClipBox_modelCAD_tlessRealPbrSO/cfg_1_to_all.py

+from mmcv import Config
+import os.path as osp
+import os
+from tqdm import tqdm
+
+cur_dir = osp.normpath(osp.dirname(osp.abspath(__file__)))
+
+base_cfg_name = "1.py"
+base_obj_name = "1"
+
+# -----------------------------------------------------------------
+id2obj = {i: str(i) for i in range(1, 31)}
+obj2id = {_name: _id for _id, _name in id2obj.items()}
+
+
+def main():
+    base_cfg_path = osp.join(cur_dir, base_cfg_name)
+    assert osp.exists(base_cfg_path), base_cfg_path  # make sure base cfg is in this dir
+    cfg = Config.fromfile(base_cfg_path)
+
+    for obj_id, obj_name in tqdm(id2obj.items()):
+        print(obj_name)
+        if obj_name == base_obj_name:
+            continue
+        # NOTE: what fields should be updated ---------------------------
+        new_cfg_dict = dict(
+            _base_="./{}".format(base_cfg_name),
+            OUTPUT_DIR=cfg.OUTPUT_DIR.replace(base_obj_name, obj_name),
+            DATASETS=dict(
+                TRAIN=("tless_{}_train_pbr".format(obj_name),),
+                TRAIN2=("tless_{}_train_primesense".format(obj_name),),
+            ),
+        )
+        # ----------------------------------------------------------------------
+        new_cfg_path = osp.join(cur_dir, base_cfg_name.replace(base_obj_name, obj_name))
+        if osp.exists(new_cfg_path):
+            raise RuntimeError("new cfg exists!")
+        new_cfg = Config(new_cfg_dict)
+        with open(new_cfg_path, "w") as f:
+            f.write(new_cfg.pretty_text)
+
+    # re-format
+    os.system("black -l 120 {}".format(cur_dir))
+
+
+if __name__ == "__main__":
+    main()