add dataset with labels start from 1

DariaMityagina · Sep 10, 2020 · 2a6d45a · 2a6d45a
1 parent 1e35a61
commit 2a6d45a
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/models/intel/vehicle-detection-0203/accuracy-check.yml b/models/intel/vehicle-detection-0203/accuracy-check.yml
@@ -7,7 +7,7 @@ models:
            type: class_agnostic_detection
 
     datasets:
-      - name: crossroad_extra_untagged_vehicle
+      - name: crossroad_extra_untagged_vehicle_labels_from_1
         preprocessing:
           - type: resize
             dst_width: 1344

diff --git a/models/intel/vehicle-detection-0204/accuracy-check.yml b/models/intel/vehicle-detection-0204/accuracy-check.yml
@@ -6,7 +6,7 @@ models:
         adapter:
            type: class_agnostic_detection
     datasets:
-      - name: crossroad_extra_untagged_vehicle
+      - name: crossroad_extra_untagged_vehicle_labels_from_1
         preprocessing:
           - type: resize
             dst_width: 1344

diff --git a/models/intel/vehicle-detection-0205/accuracy-check.yml b/models/intel/vehicle-detection-0205/accuracy-check.yml
@@ -6,7 +6,7 @@ models:
         adapter:
            type: class_agnostic_detection
     datasets:
-      - name: crossroad_extra_untagged_vehicle
+      - name: crossroad_extra_untagged_vehicle_labels_from_1
         preprocessing:
           - type: resize
             dst_width: 1344

diff --git a/tools/accuracy_checker/dataset_definitions.yml b/tools/accuracy_checker/dataset_definitions.yml
@@ -402,7 +402,7 @@ datasets:
       use_full_label_map: False
     annotation: crossroad_extra_untagged_person_hb.pickle
     dataset_meta: crossroad_extra_untagged_person_hb.json
-
+  
   - name: crossroad_extra_untagged_vehicle
     data_source: crossroad_extra_untagged/val_data
     annotation_conversion:
@@ -413,7 +413,16 @@ datasets:
       use_full_label_map: False
     annotation: crossroad_extra_untagged_vehicle.pickle
     dataset_meta: crossroad_extra_untagged_vehicle.json
-
+
+  - name: crossroad_extra_untagged_vehicle_labels_from_1
+    data_source: crossroad_extra_untagged/val_data
+    annotation_conversion:
+    converter: mscoco_detection
+    annotation_file: crossroad_extra_untagged/val_coco_no_bg_vehicle.json
+    has_background: True
+    annotation: crossroad_extra_untagged_vehicle_labels_from_1.pickle
+    dataset_meta: crossroad_extra_untagged_vehicle_labels_from_1.json
+
   - name: pedestrian_and_vehicle_dataset
     data_source: val2017
     annotation_conversion: