From 2a6d45a51f52f2aded662440336eb770ba4941a3 Mon Sep 17 00:00:00 2001 From: Gorunov Date: Thu, 10 Sep 2020 13:30:26 +0300 Subject: [PATCH] add dataset with labels start from 1 --- .../intel/vehicle-detection-0203/accuracy-check.yml | 2 +- .../intel/vehicle-detection-0204/accuracy-check.yml | 2 +- .../intel/vehicle-detection-0205/accuracy-check.yml | 2 +- tools/accuracy_checker/dataset_definitions.yml | 13 +++++++++++-- 4 files changed, 14 insertions(+), 5 deletions(-) diff --git a/models/intel/vehicle-detection-0203/accuracy-check.yml b/models/intel/vehicle-detection-0203/accuracy-check.yml index a62c0aa0829..db3c75cb2af 100644 --- 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 index 60c9b301aae..4670f974ce6 100644 --- 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 index fb021c3cce6..ce289c2d19a 100644 --- 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 index 9f16127f489..10f889aa4f5 100644 --- 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: