【发布时间】:2020-07-09 17:16:46
【问题描述】:
我需要实时使用一些物体检测算法(会在驾驶时检测物体)。 到目前为止,我已经设法在 GPU 电脑上使用 YOLO-v3 和 pytorch 并取得了很好的效果。
有没有一种算法可以在笔记本电脑的 ~1.6GHz CPU 上使用? 我当前的网络在 CPU 上检测到每帧约 5 秒,这对于实时来说太慢了......
提前致谢
【问题讨论】:
-
是的,绝对是,一台具有高速互联网的电脑,并将其与 Kaggle、Google Colab 或从 Google VM 租用张量工作站相结合。 Yolov3 是目前执行几乎即时对象识别的最有效算法。除非您花额外的钱购买 GTX 1060 及更高版本,否则拥有 1.6 Ghz 并不能带您走多远。
-
是的,您可以使用 YOLO 和 tensorflow 在您的 CPU 上进行计算。但这不会像在 GPU 上那样好。但可能会有延迟,你的 GPU 以获得更好的性能。
标签: tensorflow deep-learning neural-network pytorch yolo