From cd05d70058e4dbb5507edf539d2cc5aba4c7099d Mon Sep 17 00:00:00 2001
From: Feng Ni <nemonameless@qq.com>
Date: Tue, 7 Jun 2022 13:27:57 +0800
Subject: [PATCH] fix ppyoloe bytetrack configs (#6136)

* fix ppyoloe cfgs

* revert xpu cfgs
---
 .../_base_/ppyoloe_mot_reader_640x640.yml      | 18 +++++++++++-------
 configs/mot/bytetrack/bytetrack_ppyoloe.yml    |  3 +--
 .../bytetrack/bytetrack_ppyoloe_pplcnet.yml    |  1 -
 .../ppyoloe_crn_l_36e_640x640_mot17half.yml    |  6 ++++--
 4 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/configs/mot/bytetrack/_base_/ppyoloe_mot_reader_640x640.yml b/configs/mot/bytetrack/_base_/ppyoloe_mot_reader_640x640.yml
index c1e7ab841..ef6342fd0 100644
--- a/configs/mot/bytetrack/_base_/ppyoloe_mot_reader_640x640.yml
+++ b/configs/mot/bytetrack/_base_/ppyoloe_mot_reader_640x640.yml
@@ -1,4 +1,8 @@
-worker_num: 8
+worker_num: 4
+eval_height: &eval_height 640
+eval_width: &eval_width 640
+eval_size: &eval_size [*eval_height, *eval_width]
+
 TrainReader:
   sample_transforms:
     - Decode: {}
@@ -20,17 +24,17 @@ TrainReader:
 EvalReader:
   sample_transforms:
     - Decode: {}
-    - Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
+    - Resize: {target_size: *eval_size, keep_ratio: False, interp: 2}
     - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
     - Permute: {}
   batch_size: 8
 
 TestReader:
   inputs_def:
-    image_shape: [3, 640, 640]
+    image_shape: [3, *eval_height, *eval_width]
   sample_transforms:
     - Decode: {}
-    - Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
+    - Resize: {target_size: *eval_size, keep_ratio: False, interp: 2}
     - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
     - Permute: {}
   batch_size: 1
@@ -40,17 +44,17 @@ TestReader:
 EvalMOTReader:
   sample_transforms:
     - Decode: {}
-    - Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
+    - Resize: {target_size: *eval_size, keep_ratio: False, interp: 2}
     - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
     - Permute: {}
   batch_size: 1
 
 TestMOTReader:
   inputs_def:
-    image_shape: [3, 640, 640]
+    image_shape: [3, *eval_height, *eval_width]
   sample_transforms:
     - Decode: {}
-    - Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
+    - Resize: {target_size: *eval_size, keep_ratio: False, interp: 2}
     - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
     - Permute: {}
   batch_size: 1
diff --git a/configs/mot/bytetrack/bytetrack_ppyoloe.yml b/configs/mot/bytetrack/bytetrack_ppyoloe.yml
index 08b7a00d8..5e7ffe07f 100644
--- a/configs/mot/bytetrack/bytetrack_ppyoloe.yml
+++ b/configs/mot/bytetrack/bytetrack_ppyoloe.yml
@@ -8,7 +8,7 @@ weights: output/bytetrack_ppyoloe/model_final
 log_iter: 20
 snapshot_epoch: 2
 
-metric: MOT # eval/infer mode
+metric: MOT # eval/infer mode, set 'COCO' can be training mode
 num_classes: 1
 
 architecture: ByteTrack
@@ -33,7 +33,6 @@ PPYOLOEHead:
   grid_cell_offset: 0.5
   static_assigner_epoch: -1 # 100
   use_varifocal_loss: True
-  eval_input_size: [640, 640]
   loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
   static_assigner:
     name: ATSSAssigner
diff --git a/configs/mot/bytetrack/bytetrack_ppyoloe_pplcnet.yml b/configs/mot/bytetrack/bytetrack_ppyoloe_pplcnet.yml
index 98ea15f15..60f81165d 100644
--- a/configs/mot/bytetrack/bytetrack_ppyoloe_pplcnet.yml
+++ b/configs/mot/bytetrack/bytetrack_ppyoloe_pplcnet.yml
@@ -33,7 +33,6 @@ PPYOLOEHead:
   grid_cell_offset: 0.5
   static_assigner_epoch: -1 # 100
   use_varifocal_loss: True
-  eval_input_size: [640, 640]
   loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
   static_assigner:
     name: ATSSAssigner
diff --git a/configs/mot/bytetrack/detector/ppyoloe_crn_l_36e_640x640_mot17half.yml b/configs/mot/bytetrack/detector/ppyoloe_crn_l_36e_640x640_mot17half.yml
index 89654f059..6c770e9bf 100644
--- a/configs/mot/bytetrack/detector/ppyoloe_crn_l_36e_640x640_mot17half.yml
+++ b/configs/mot/bytetrack/detector/ppyoloe_crn_l_36e_640x640_mot17half.yml
@@ -7,6 +7,7 @@ weights: output/ppyoloe_crn_l_36e_640x640_mot17half/model_final
 log_iter: 20
 snapshot_epoch: 2
 
+
 # schedule configuration for fine-tuning
 epoch: 36
 LearningRate:
@@ -16,7 +17,7 @@ LearningRate:
       max_epochs: 43
     - !LinearWarmup
       start_factor: 0.001
-      steps: 100
+      epochs: 1
 
 OptimizerBuilder:
   optimizer:
@@ -26,9 +27,11 @@ OptimizerBuilder:
     factor: 0.0005
     type: L2
 
+
 TrainReader:
   batch_size: 8
 
+
 # detector configuration
 architecture: YOLOv3
 norm_type: sync_bn
@@ -63,7 +66,6 @@ PPYOLOEHead:
   grid_cell_offset: 0.5
   static_assigner_epoch: -1 # 100
   use_varifocal_loss: True
-  eval_input_size: [640, 640]
   loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
   static_assigner:
     name: ATSSAssigner
-- 
GitLab