0

我已经使用AlexeyAB 的darknet 训练了一个自定义的 YoloV4 单类对象检测器。

我的模型的最佳权重文件导致以下性能

我在 google colab 上训练的模型的指标

我使用命令获得了这个评估,

darknet.exe detector map data/obj.data yolo-obj.cfg backup\yolo-custom-best.weights

在视频上运行模型后,我得到大约 25 fps。

我的问题是:当我阅读有关 YOLO 的文章时,他们声称它以毫秒 (ms) 为单位进行检测。为什么我的模型显示检测时间为 5 秒,但在 FPS 上与其他更快的模型相当?

  • 图像中显示的检测时间(由给定命令返回)是否有所不同?(如果是,那么我如何找到我的模型的速度?)
  • 如果否,如何使其更快?
4

1 回答 1

1

请注意,论文中提到的检测时间与此处的 GitHub 存储库中提到的检测时间相同,是假设它在GPU上的single batch测试时间运行。Tesla V100除了这些因素之外,运行时间还根据输入图像的大小而变化

如果您使用不同的 GPU 或 CPU,发现更长的检测时间也就不足为奇了。FPS也是如此

于 2021-04-20T02:03:15.567 回答