python调用检测模型输出图片检测结果和基于C++通过libtorch输出结果是一致的吗？

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输出结果是一致的吗？ #6

Open litc-rgb opened 2 years ago

litc-rgb commented 2 years ago

您分享的代码非常有用，感谢您的分享。关于您分享的代码，有个问题需要向您确认：
您使用torchvision自带的nms时，python调用检测模型输出图片检测结果和基于C++通过libtorch输出结果是一致的吗？
我们重写了nms，python调用检测模型输出图片检测结果和基于C++通过libtorch输出结果是不一致的。请问：这个必须使用torchvision自带的nms吗？谢谢~