search

miemie2013 / miemiedetection

Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.
Apache License 2.0
287 stars 34 forks source link

ppyoloe_crn_s_300e_coco.pth速度比yolov5s.pt慢很多 #6

Open XiaoJiNu opened 2 years ago

XiaoJiNu commented 2 years ago

您好： ppyoloe_crn_s_300e_coco.pth在640x640分辨率下测试比yolov5s.pt慢很多，各自的速度分别为5ms和9ms左右。 我的环境是3060ti, pytorch10。 请问您对比过吗

miemie2013 commented 2 years ago

您好： ppyoloe_crn_s_300e_coco.pth在640x640分辨率下测试比yolov5s.pt慢很多，各自的速度分别为5ms和9ms左右。 我的环境是3060ti, pytorch10。 请问您对比过吗

我没有试过yolov5，请问您有仔细比较二者的差异吗？比如yolov5测速时是否不包括前处理、后处理（box解码和nms）的时间？yolov5是否合并了卷积层和bn层？yolov5是否使用了fp16进行预测？两个仓库预测时的批大小是否都一致？两个仓库上述条件一致才好比较。另外，你可以将self.eval_data_num_workers设置得大一点来加速。我默认写小了是为了照顾电脑配置不够高的用户。