search

PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Apache License 2.0
12.82k stars 2.89k forks source link

Yolov5 后处理层不支持 #3818

Closed youngstu closed 2 years ago

youngstu commented 3 years ago

问题描述： 1、pytorch训练模型，导出onnx，再转到paddle，paddle模型没有实现yolov5后处理部分（解码+nms），请问怎么实现呢？

heavengate commented 3 years ago

nms我们有MultiClassNMS，解码部分的话跟这个YOLOv3Head部分应该是基本一致的，如果要用的话可以结合这个组一下 https://github.com/PaddlePaddle/PaddleDetection/blob/f53e8f7f48d5f952d487da8ea75f6793086156da/ppdet/modeling/heads/yolo_head.py#L17 另外PaddleDetection里开源了PP-YOLOv2模型，精度速度也是由于yolov5，也可以试下这个

paddle-bot-old[bot] commented 2 years ago

Since this issue has not been updated for more than three months, it will be closed, if it is not solved or there is a follow-up one, please reopen it at any time and we will continue to follow up. It is recommended to pull and try the latest code first. 由于该问题超过三个月未更新，将会被关闭，若问题未解决或有后续问题，请随时重新打开（建议先拉取最新代码进行尝试），我们会继续跟进。