From c08133674d9b6aa3b33cbad75c614290e4a4e541 Mon Sep 17 00:00:00 2001
From: osimeoni <oriane.simeoni@valeo.com>
Date: Fri, 21 Jan 2022 12:42:55 +0000
Subject: [PATCH] small fix - CAD classes
---
...are_voc_LOST_CAD_pseudo_boxes_in_detectron2_format.py | 2 +-
tools/train_net_for_LOST_CAD.py | 9 +++------
2 files changed, 4 insertions(+), 7 deletions(-)
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 4bc5183..22b528a 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 2dcd088..cf63451 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
--
GitLab