未验证 提交 50da62fd 编写于 作者: G Guanghua Yu 提交者: GitHub

fix picodet_pedestrian export (#5395)

上级 5471a176
...@@ -12,6 +12,11 @@ snapshot_epoch: 10 ...@@ -12,6 +12,11 @@ snapshot_epoch: 10
epoch: 300 epoch: 300
metric: COCO metric: COCO
num_classes: 1 num_classes: 1
# Exporting the model
export:
post_process: False # Whether post-processing is included in the network when export model.
nms: False # Whether NMS is included in the network when export model.
benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
architecture: PicoDet architecture: PicoDet
......
...@@ -12,6 +12,11 @@ snapshot_epoch: 10 ...@@ -12,6 +12,11 @@ snapshot_epoch: 10
epoch: 300 epoch: 300
metric: COCO metric: COCO
num_classes: 1 num_classes: 1
# Exporting the model
export:
post_process: False # Whether post-processing is included in the network when export model.
nms: False # Whether NMS is included in the network when export model.
benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
architecture: PicoDet architecture: PicoDet
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册