Update templates for person detection

configs/ote/horizontal-text-detection/horizontal-text-detection-0001/compression_config.json

@@ -13,11 +13,9 @@
         "optimizer": {
             "lr": 0.0001
         },
-        "lr_config": {
-            "_delete_": true,
-            "policy": "fixed"
+        "runner": {
+            "max_epochs": 2
         },
-        "total_epochs": 2,
         "nncf_config": {
             "compression": [
                 {

configs/ote/horizontal-text-detection/horizontal-text-detection-0001/model.py

@@ -50,7 +50,7 @@
         score_thr=0.25,
         nms=dict(type='nms', iou_threshold=0.5),
         max_per_img=100))
-evaluation = dict(interval=1, metric=['bbox', 'f1'], score_thr=0.25)
+evaluation = dict(interval=1, metric='mAP')
 # dataset settings
 dataset_type = 'CocoDataset'
 data_root = 'data/text-dataset/'
@@ -140,11 +140,11 @@
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 25
+runner = dict(type='EpochBasedRunner', max_epochs=25)
 device_ids = range(4)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
-work_dir = 'outputs/horizontal-text-detection'
-load_from = None
+work_dir = 'output'
+load_from = 'https://download.01.org/opencv/openvino_training_extensions/models/object_detection/v2/horizontal-text-detection-0001.pth'
 resume_from = None
 workflow = [('train', 1)]

configs/ote/horizontal-text-detection/horizontal-text-detection-0001/template_experimental.yaml

@@ -0,0 +1,57 @@
+# Description.
+model_template_id: Horizontal_Text_Detection_0001
+name: horizontal-text-detection
+task_type: DETECTION
+task_family: VISION
+instantiation: "CLASS"
+summary: Horizontal Text Detection model (MobileNetV2-FCOS).
+application:
+  ~
+
+# Algo backend.
+framework: OTEDetection v2.9.1
+
+# Task implementations.
+entrypoints:
+  base: mmdet.apis.ote.apis.detection.OTEDetectionTrainingTask
+  openvino: mmdet.apis.ote.apis.detection.OpenVINODetectionTask
+  nncf: mmdet.apis.ote.apis.detection.OTEDetectionNNCFTask
+
+# Capabilities.
+capabilities:
+  - compute_representations
+
+# Hyperparameters.
+hyper_parameters:
+  base_path: ../../../../mmdet/apis/ote/apis/detection/configuration.yaml
+  parameter_overrides:
+    learning_parameters:
+      batch_size:
+        default_value: 8
+      # LR reduction proportionally to the batch size decrease (because gpu_num=1)
+      learning_rate:
+        default_value: 0.005
+      learning_rate_warmup_iters:
+        default_value: 500
+      num_iters:
+        default_value: 25
+    nncf_optimization:
+      enable_quantization:
+        default_value: false
+      enable_pruning:
+        default_value: false
+      maximal_accuracy_degradation:
+        default_value: 0.01
+
+# Training resources.
+max_nodes: 1
+training_targets:
+  - GPU
+  - CPU
+
+# Stats.
+gigaflops: 8.0 #to be checked with model-analyzer
+size: 1.95
+
+# Obsolete
+# gpu_nums: 2

configs/ote/person-detection/person-detection-0200/compression_config.json

@@ -21,11 +21,9 @@
             "momentum": 0.9,
             "weight_decay": 0.0005
         },
-        "lr_config": {
-            "_delete_": true,
-            "policy": "fixed"
+        "runner": {
+            "max_epochs": 2
         },
-        "total_epochs": 2,
         "nncf_config": {
             "compression": [
                 {

configs/ote/person-detection/person-detection-0200/model.py

@@ -152,10 +152,10 @@
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 20
+runner = dict(type='EpochBasedRunner', max_epochs=20)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
-work_dir = 'outputs/person-detection-0200'
-load_from = None
+work_dir = 'output'
+load_from = 'https://download.01.org/opencv/openvino_training_extensions/models/object_detection/v2/person-detection-0200-1.pth'
 resume_from = None
-workflow = [('train', 1)]
+workflow = [('train', 1)]

configs/ote/person-detection/person-detection-0200/template_experimental.yaml

@@ -0,0 +1,57 @@
+# Description.
+model_template_id: Person_Detection_0200
+name: person-detection-0200
+task_type: DETECTION
+task_family: VISION
+instantiation: "CLASS"
+summary: Fastest Person Detection model for large and simple objects (MobileNetV2-SSD).
+application:
+  ~
+
+# Algo backend.
+framework: OTEDetection v2.9.1
+
+# Task implementations.
+entrypoints:
+  base: mmdet.apis.ote.apis.detection.OTEDetectionTrainingTask
+  openvino: mmdet.apis.ote.apis.detection.OpenVINODetectionTask
+  nncf: mmdet.apis.ote.apis.detection.OTEDetectionNNCFTask
+
+# Capabilities.
+capabilities:
+  - compute_representations
+
+# Hyperparameters.
+hyper_parameters:
+  base_path: ../../../../mmdet/apis/ote/apis/detection/configuration.yaml
+  parameter_overrides:
+    learning_parameters:
+      batch_size:
+        default_value: 122
+      # LR reduction proportionally to the batch size decrease (because gpu_num=1)
+      learning_rate:
+        default_value: 0.025
+      learning_rate_warmup_iters:
+        default_value: 1200
+      num_iters:
+        default_value: 20
+    nncf_optimization:
+      enable_quantization:
+        default_value: false
+      enable_pruning:
+        default_value: false
+      maximal_accuracy_degradation:
+        default_value: 0.01
+
+# Training resources.
+max_nodes: 1
+training_targets:
+  - GPU
+  - CPU
+
+# Stats.
+gigaflops: 0.82 #to be checked with model-analyzer
+size: 1.83
+
+# Obsolete
+# gpu_nums: 2

configs/ote/person-detection/person-detection-0201/compression_config.json

@@ -21,11 +21,9 @@
             "momentum": 0.9,
             "weight_decay": 0.0005
         },
-        "lr_config": {
-            "_delete_": true,
-            "policy": "fixed"
+        "runner": {
+            "max_epochs": 2
         },
-        "total_epochs": 2,
         "nncf_config": {
             "compression": [
                 {

configs/ote/person-detection/person-detection-0201/model.py

@@ -152,10 +152,10 @@
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 20
+runner = dict(type='EpochBasedRunner', max_epochs=20)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
-work_dir = 'outputs/person-detection-0201'
-load_from = None
+work_dir = 'output'
+load_from = 'https://download.01.org/opencv/openvino_training_extensions/models/object_detection/v2/person-detection-0201-1.pth'
 resume_from = None
-workflow = [('train', 1)]
+workflow = [('train', 1)]

configs/ote/person-detection/person-detection-0201/template_experimental.yaml

@@ -0,0 +1,57 @@
+# Description.
+model_template_id: Person_Detection_0201
+name: person-detection-0201
+task_type: DETECTION
+task_family: VISION
+instantiation: "CLASS"
+summary: Fast Person Detection models (MobileNetV2-SSD).
+application:
+  ~
+
+# Algo backend.
+framework: OTEDetection v2.9.1
+
+# Task implementations.
+entrypoints:
+  base: mmdet.apis.ote.apis.detection.OTEDetectionTrainingTask
+  openvino: mmdet.apis.ote.apis.detection.OpenVINODetectionTask
+  nncf: mmdet.apis.ote.apis.detection.OTEDetectionNNCFTask
+
+# Capabilities.
+capabilities:
+  - compute_representations
+
+# Hyperparameters.
+hyper_parameters:
+  base_path: ../../../../mmdet/apis/ote/apis/detection/configuration.yaml
+  parameter_overrides:
+    learning_parameters:
+      batch_size:
+        default_value: 84
+      # LR reduction proportionally to the batch size decrease (because gpu_num=1)
+      learning_rate:
+        default_value: 0.0125
+      learning_rate_warmup_iters:
+        default_value: 1200
+      num_iters:
+        default_value: 20
+    nncf_optimization:
+      enable_quantization:
+        default_value: false
+      enable_pruning:
+        default_value: false
+      maximal_accuracy_degradation:
+        default_value: 0.01
+
+# Training resources.
+max_nodes: 1
+training_targets:
+  - GPU
+  - CPU
+
+# Stats.
+gigaflops: 1.84 #to be checked with model-analyzer
+size: 1.83
+
+# Obsolete
+# gpu_nums: 4

configs/ote/person-detection/person-detection-0202/compression_config.json

@@ -21,11 +21,9 @@
             "momentum": 0.9,
             "weight_decay": 0.0005
         },
-        "lr_config": {
-            "_delete_": true,
-            "policy": "fixed"
+        "runner": {
+            "max_epochs": 2
         },
-        "total_epochs": 2,
         "nncf_config": {
             "compression": [
                 {

configs/ote/person-detection/person-detection-0202/model.py

@@ -152,10 +152,10 @@
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 20
+runner = dict(type='EpochBasedRunner', max_epochs=20)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
-work_dir = 'outputs/person-detection-0202'
-load_from = None
+work_dir = 'output'
+load_from = 'https://download.01.org/opencv/openvino_training_extensions/models/object_detection/v2/person-detection-0202-1.pth'
 resume_from = None
-workflow = [('train', 1)]
+workflow = [('train', 1)]

configs/ote/person-detection/person-detection-0202/template_experimental.yaml

@@ -0,0 +1,57 @@
+# Description.
+model_template_id: Person_Detection_0202
+name: person-detection-0202
+task_type: DETECTION
+task_family: VISION
+instantiation: "CLASS"
+summary: Medium Person Detection model (MobileNetV2-SSD).
+application:
+  ~
+
+# Algo backend.
+framework: OTEDetection v2.9.1
+
+# Task implementations.
+entrypoints:
+  base: mmdet.apis.ote.apis.detection.OTEDetectionTrainingTask
+  openvino: mmdet.apis.ote.apis.detection.OpenVINODetectionTask
+  nncf: mmdet.apis.ote.apis.detection.OTEDetectionNNCFTask
+
+# Capabilities.
+capabilities:
+  - compute_representations
+
+# Hyperparameters.
+hyper_parameters:
+  base_path: ../../../../mmdet/apis/ote/apis/detection/configuration.yaml
+  parameter_overrides:
+    learning_parameters:
+      batch_size:
+        default_value: 30
+      # LR reduction proportionally to the batch size decrease (because gpu_num=1)
+      learning_rate:
+        default_value: 0.025
+      learning_rate_warmup_iters:
+        default_value: 1200
+      num_iters:
+        default_value: 20
+    nncf_optimization:
+      enable_quantization:
+        default_value: false
+      enable_pruning:
+        default_value: false
+      maximal_accuracy_degradation:
+        default_value: 0.01
+
+# Training resources.
+max_nodes: 1
+training_targets:
+  - GPU
+  - CPU
+
+# Stats.
+gigaflops: 2.52 #to be checked with model-analyzer
+size: 1.83
+
+# Obsolete
+# gpu_nums: 2

configs/ote/person-detection/person-detection-0203/compression_config.json

@@ -21,11 +21,9 @@
             "momentum": 0.9,
             "weight_decay": 0.0005
         },
-        "lr_config": {
-            "_delete_": true,
-            "policy": "fixed"
+        "runner": {
+            "max_epochs": 1
         },
-        "total_epochs": 1,
         "nncf_config": {
             "compression": [
                 {

configs/ote/person-detection/person-detection-0203/model.py

@@ -148,10 +148,10 @@
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 20
+runner = dict(type='EpochBasedRunner', max_epochs=20)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
-work_dir = 'outputs/person-detection-0203'
-load_from = None
+work_dir = 'output'
+load_from = 'https://download.01.org/opencv/openvino_training_extensions/models/object_detection/v2/person-detection-0203.pth'
 resume_from = None
-workflow = [('train', 1)]
+workflow = [('train', 1)]