diff --git a/tools/prepare_voc_LOST_CAD_pseudo_boxes_in_detectron2_format.py b/tools/prepare_voc_LOST_CAD_pseudo_boxes_in_detectron2_format.py index 4bc51839c00ab8faf7d0972a0fb9f515e4bcad4e..22b528aa1a99e25686cf5620d9688c900b9b0ffb 100755 --- a/tools/prepare_voc_LOST_CAD_pseudo_boxes_in_detectron2_format.py +++ b/tools/prepare_voc_LOST_CAD_pseudo_boxes_in_detectron2_format.py @@ -84,7 +84,7 @@ if __name__ == '__main__': "name": dataset_name, "split": "trainval", "year": args.year, - "thing_classes": detectron2.data.MetadataCatalog.get(f"voc_{args.year}_trainval").thing_classes, + "thing_classes": "object", }} dst_file = f'./datasets/{dataset_name}.json' diff --git a/tools/train_net_for_LOST_CAD.py b/tools/train_net_for_LOST_CAD.py index 2dcd0882cbcb95f1c3c25cc29f60799bb9185416..cf63451f13271bec00daa5c7da53006524f21dcf 100755 --- a/tools/train_net_for_LOST_CAD.py +++ b/tools/train_net_for_LOST_CAD.py @@ -62,8 +62,7 @@ def register_voc_in_coco_style( return json_data["dataset"] detectron2.data.DatasetCatalog.register( voc2007_trainval_dataset_name, voc2007_trainval_dataset_function) - detectron2.data.MetadataCatalog.get(voc2007_trainval_dataset_name).thing_classes = ( - detectron2.data.MetadataCatalog.get("voc_2007_trainval").thing_classes) + detectron2.data.MetadataCatalog.get(voc2007_trainval_dataset_name).thing_classes = ["object",] detectron2.data.MetadataCatalog.get(voc2007_trainval_dataset_name).evaluator_type = "coco" detectron2.data.MetadataCatalog.get(voc2007_trainval_dataset_name).split = detectron2.data.MetadataCatalog.get("voc_2007_trainval").split detectron2.data.MetadataCatalog.get(voc2007_trainval_dataset_name).year = detectron2.data.MetadataCatalog.get("voc_2007_trainval").year @@ -76,8 +75,7 @@ def register_voc_in_coco_style( return json_data["dataset"] detectron2.data.DatasetCatalog.register( voc2007_test_dataset_name, voc2007_test_dataset_function) - detectron2.data.MetadataCatalog.get(voc2007_test_dataset_name).thing_classes = ( - detectron2.data.MetadataCatalog.get("voc_2007_test").thing_classes) + detectron2.data.MetadataCatalog.get(voc2007_test_dataset_name).thing_classes = ["object",] detectron2.data.MetadataCatalog.get(voc2007_test_dataset_name).evaluator_type = "coco" detectron2.data.MetadataCatalog.get(voc2007_test_dataset_name).split = detectron2.data.MetadataCatalog.get("voc_2007_test").split detectron2.data.MetadataCatalog.get(voc2007_test_dataset_name).year = detectron2.data.MetadataCatalog.get("voc_2007_test").year @@ -90,8 +88,7 @@ def register_voc_in_coco_style( return json_data["dataset"] detectron2.data.DatasetCatalog.register( voc2012_trainval_dataset_name, voc2012_trainval_dataset_function) - detectron2.data.MetadataCatalog.get(voc2012_trainval_dataset_name).thing_classes = ( - detectron2.data.MetadataCatalog.get("voc_2012_trainval").thing_classes) + detectron2.data.MetadataCatalog.get(voc2012_trainval_dataset_name).thing_classes = ["object",] detectron2.data.MetadataCatalog.get(voc2012_trainval_dataset_name).evaluator_type = "coco" detectron2.data.MetadataCatalog.get(voc2012_trainval_dataset_name).split = detectron2.data.MetadataCatalog.get("voc_2012_trainval").split detectron2.data.MetadataCatalog.get(voc2012_trainval_dataset_name).year = detectron2.data.MetadataCatalog.get("voc_2012_trainval").year