xinhappy1

WongKinYiu/PyTorch_YOLOv4-tiny，自己数据集，类别数为1，推理加速后目标上有很多框，该怎么处理？

10

您好！我使用您的方法[yolov4-tiny-tensorrt](https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT/tree/master/yolov4-tiny-tensorrt#yolov4-tiny-tensorrt) 、您的百度网盘里的pt 和 cfg 文件，可以生成 wts、engine，执行 ./yolov4-tiny -d ../samples 命令后生成的图片正常，**每个目标上只有一个框**。 但当我使用训练好的[WongKinYiu/PyTorch_YOLOv4-tiny](https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/master)时，使用我的 pt和cfg文件生成wts、engine后，对我自己的部分数据集图片进行推理时，生成的图片上**每个目标会有多个框**，就像这样 [picture](https://github.com/happyboy688/picture/blob/main/_1.jpeg)。我该如何做？ 期待您的回复！