reformat

configs/gdrn/hb_pbr/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_hb.py

@@ -32,7 +32,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 30
+    TOTAL_EPOCHS=40,  # 30
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,
@@ -148,4 +148,3 @@ TEST = dict(
     VIS=False,
     TEST_BBOX_TYPE="est",
 )  # gt | est
-

configs/gdrn/icbinPbrSO/convnext_AugCosyAAEGray_DMask_amodalClipBox_icbin/01_coffee_cup.py

@@ -32,7 +32,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=36,
-    TOTAL_EPOCHS=100, # 15
+    TOTAL_EPOCHS=100,  # 15
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,

configs/gdrn/icbin_pbr/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_icbin.py

@@ -32,7 +32,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 10
+    TOTAL_EPOCHS=40,  # 10
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,

configs/gdrn/itodd_pbr/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_itodd.py

@@ -36,7 +36,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 25
+    TOTAL_EPOCHS=40,  # 25
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,

configs/gdrn/lmo_pbr/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_lmo.py

@@ -32,7 +32,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 30
+    TOTAL_EPOCHS=40,  # 30
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,
@@ -140,4 +140,3 @@ VAL = dict(
 )
 
 TEST = dict(EVAL_PERIOD=0, VIS=False, TEST_BBOX_TYPE="est")  # gt | est
-

configs/gdrn/tless/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_tless.py

@@ -33,7 +33,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 30
+    TOTAL_EPOCHS=40,  # 30
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,
@@ -44,7 +44,10 @@ SOLVER = dict(
 )
 
 DATASETS = dict(
-    TRAIN=("tless_train_primesense", "tless_train_pbr",),
+    TRAIN=(
+        "tless_train_primesense",
+        "tless_train_pbr",
+    ),
     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",),

configs/gdrn/ycbv/convnext_a6_AugCosyAAEGray_BG05_mlL1_DMask_amodalClipBox_classAware_ycbv.py

@@ -33,7 +33,7 @@ INPUT = dict(
 
 SOLVER = dict(
     IMS_PER_BATCH=48,
-    TOTAL_EPOCHS=40, # 10
+    TOTAL_EPOCHS=40,  # 10
     LR_SCHEDULER_NAME="flat_and_anneal",
     ANNEAL_METHOD="cosine",  # "cosine"
     ANNEAL_POINT=0.72,

core/gdrn_modeling/tools/convert_det_to_our_format_1b.py

@@ -5,12 +5,18 @@ import json
 import os
 
 parser = argparse.ArgumentParser(description="convert det from bop format to ours")
-parser.add_argument("--idir", type=str, default="/data1/tangjw/projects/YOLOX_benckmark/master/object_pose_benchmark/output/yolox/bop_pbr/multi_scale_test", help="input path")
+parser.add_argument(
+    "--idir",
+    type=str,
+    default="/data1/tangjw/projects/YOLOX_benckmark/master/object_pose_benchmark/output/yolox/bop_pbr/multi_scale_test",
+    help="input path",
+)
 parser.add_argument("--odir", type=str, default="datasets/BOP_DATASETS/", help="output path")
 args = parser.parse_args()
 
 dataset_names = ["hb", "icbin", "itodd", "lmo", "tless", "tudl", "ycbv"]
 
+
 def convert_format(ipath, opath):
     ds = mmcv.load(ipath)
     outs = {}
@@ -80,8 +86,8 @@ if __name__ == "__main__":
     for json_file in json_files:
         for dset in dataset_names:
             if dset in json_file:
-                break 
-    
+                break
+
         ipath = os.path.join(args.idir, json_file)
 
         odir = os.path.join(args.odir, dset, "test/test_bboxes")

ref/tless.py

@@ -23,7 +23,7 @@ train_render_dir = osp.join(dataset_root, "train_render_reconst")
 test_dir = osp.join(dataset_root, "test_primesense")
 
 # model_dir = osp.join(dataset_root, "models_reconst")  # use recon models as default
-model_dir = osp.join(dataset_root, "models_cad") 
+model_dir = osp.join(dataset_root, "models_cad")
 model_cad = osp.join(dataset_root, "models_cad")
 model_reconst_dir = osp.join(dataset_root, "models_reconst")
 model_eval_dir = osp.join(dataset_root, "models_eval")