未验证 提交 cd05d700 编写于 作者: F Feng Ni 提交者: GitHub

fix ppyoloe bytetrack configs (#6136)

* fix ppyoloe cfgs

* revert xpu cfgs
上级 9c689a4b
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: TrainReader:
sample_transforms: sample_transforms:
- Decode: {} - Decode: {}
...@@ -20,17 +24,17 @@ TrainReader: ...@@ -20,17 +24,17 @@ TrainReader:
EvalReader: EvalReader:
sample_transforms: sample_transforms:
- Decode: {} - 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} - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- Permute: {} - Permute: {}
batch_size: 8 batch_size: 8
TestReader: TestReader:
inputs_def: inputs_def:
image_shape: [3, 640, 640] image_shape: [3, *eval_height, *eval_width]
sample_transforms: sample_transforms:
- Decode: {} - 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} - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- Permute: {} - Permute: {}
batch_size: 1 batch_size: 1
...@@ -40,17 +44,17 @@ TestReader: ...@@ -40,17 +44,17 @@ TestReader:
EvalMOTReader: EvalMOTReader:
sample_transforms: sample_transforms:
- Decode: {} - 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} - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- Permute: {} - Permute: {}
batch_size: 1 batch_size: 1
TestMOTReader: TestMOTReader:
inputs_def: inputs_def:
image_shape: [3, 640, 640] image_shape: [3, *eval_height, *eval_width]
sample_transforms: sample_transforms:
- Decode: {} - 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} - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- Permute: {} - Permute: {}
batch_size: 1 batch_size: 1
...@@ -8,7 +8,7 @@ weights: output/bytetrack_ppyoloe/model_final ...@@ -8,7 +8,7 @@ weights: output/bytetrack_ppyoloe/model_final
log_iter: 20 log_iter: 20
snapshot_epoch: 2 snapshot_epoch: 2
metric: MOT # eval/infer mode metric: MOT # eval/infer mode, set 'COCO' can be training mode
num_classes: 1 num_classes: 1
architecture: ByteTrack architecture: ByteTrack
...@@ -33,7 +33,6 @@ PPYOLOEHead: ...@@ -33,7 +33,6 @@ PPYOLOEHead:
grid_cell_offset: 0.5 grid_cell_offset: 0.5
static_assigner_epoch: -1 # 100 static_assigner_epoch: -1 # 100
use_varifocal_loss: True use_varifocal_loss: True
eval_input_size: [640, 640]
loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5} loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
static_assigner: static_assigner:
name: ATSSAssigner name: ATSSAssigner
......
...@@ -33,7 +33,6 @@ PPYOLOEHead: ...@@ -33,7 +33,6 @@ PPYOLOEHead:
grid_cell_offset: 0.5 grid_cell_offset: 0.5
static_assigner_epoch: -1 # 100 static_assigner_epoch: -1 # 100
use_varifocal_loss: True use_varifocal_loss: True
eval_input_size: [640, 640]
loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5} loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
static_assigner: static_assigner:
name: ATSSAssigner name: ATSSAssigner
......
...@@ -7,6 +7,7 @@ weights: output/ppyoloe_crn_l_36e_640x640_mot17half/model_final ...@@ -7,6 +7,7 @@ weights: output/ppyoloe_crn_l_36e_640x640_mot17half/model_final
log_iter: 20 log_iter: 20
snapshot_epoch: 2 snapshot_epoch: 2
# schedule configuration for fine-tuning # schedule configuration for fine-tuning
epoch: 36 epoch: 36
LearningRate: LearningRate:
...@@ -16,7 +17,7 @@ LearningRate: ...@@ -16,7 +17,7 @@ LearningRate:
max_epochs: 43 max_epochs: 43
- !LinearWarmup - !LinearWarmup
start_factor: 0.001 start_factor: 0.001
steps: 100 epochs: 1
OptimizerBuilder: OptimizerBuilder:
optimizer: optimizer:
...@@ -26,9 +27,11 @@ OptimizerBuilder: ...@@ -26,9 +27,11 @@ OptimizerBuilder:
factor: 0.0005 factor: 0.0005
type: L2 type: L2
TrainReader: TrainReader:
batch_size: 8 batch_size: 8
# detector configuration # detector configuration
architecture: YOLOv3 architecture: YOLOv3
norm_type: sync_bn norm_type: sync_bn
...@@ -63,7 +66,6 @@ PPYOLOEHead: ...@@ -63,7 +66,6 @@ PPYOLOEHead:
grid_cell_offset: 0.5 grid_cell_offset: 0.5
static_assigner_epoch: -1 # 100 static_assigner_epoch: -1 # 100
use_varifocal_loss: True use_varifocal_loss: True
eval_input_size: [640, 640]
loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5} loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
static_assigner: static_assigner:
name: ATSSAssigner name: ATSSAssigner
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册