wsx000 / YOLOv4-LibTorch-GPU-deployment

This is a YOLOv4 post-processing code used gpu based on pure libtorch API
GNU General Public License v2.0
14 stars 7 forks source link

python调用模型输出图片检测结果和基于C++通过libtorch输出结果不一致 #5

Open litc-rgb opened 2 years ago

litc-rgb commented 2 years ago

问题描述: 需要向您确认:您使用torchvision自带的nms时,python调用检测模型输出图片检测结果和基于C++通过libtorch输出结果是一致的吗? 我们重写了nms,python调用检测模型输出图片检测结果和基于C++通过libtorch输出结果是不一致的。请问:这个必须使用torchvision自带的nms吗?谢谢~

wsx000 commented 2 years ago

时间久了我也记不清是不是一样的了,但是官方给的nms只是一种实现的API接口,nms原理都是通用的,自己实现只要确认结果没错也是没问题的,不一定必须用官方的库函数~

litc-rgb commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。你好,我最近正在休假中,无法亲自回复你的邮件。我将在假期结束后,尽快给你回复。