yolox中关于NMS你需要知道的
最新推荐文章于 2023-10-12 20:55:36 发布
这篇博客探讨了YOLOX模型在不同阶段使用非极大值抑制(NMS)函数的区别。在demo.py中,模型采用torchvision.ops.nms()进行不区分类别的NMS操作。而在train.py和eval.py中,为了考虑类别信息,模型使用torchvision.ops.batched_nms()。这两种NMS函数的使用体现了在目标检测任务中对精度和效率的不同需求。
摘要由CSDN通过智能技术生成