Skip to content

Commit 50da62f

Browse files
authored
fix picodet_pedestrian export (PaddlePaddle#5395)
1 parent 5471a17 commit 50da62f

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

configs/picodet/application/pedestrian_detection/picodet_s_192_pedestrian.yml

+5
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@ snapshot_epoch: 10
1212
epoch: 300
1313
metric: COCO
1414
num_classes: 1
15+
# Exporting the model
16+
export:
17+
post_process: False # Whether post-processing is included in the network when export model.
18+
nms: False # Whether NMS is included in the network when export model.
19+
benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
1520

1621
architecture: PicoDet
1722

configs/picodet/application/pedestrian_detection/picodet_s_320_pedestrian.yml

+5
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@ snapshot_epoch: 10
1212
epoch: 300
1313
metric: COCO
1414
num_classes: 1
15+
# Exporting the model
16+
export:
17+
post_process: False # Whether post-processing is included in the network when export model.
18+
nms: False # Whether NMS is included in the network when export model.
19+
benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
1520

1621
architecture: PicoDet
1722

0 commit comments

Comments
 (0)