我们可以用CPU代替GPU来训练目标检测的YOLO模型吗 [英] Can we use CPU instead of GPU to train custome YOLO model for object detection

问题描述

我想为我的定制对象数据集训练YOLO模型。我在不同的网站上到处看到它，每个人都在谈论应该使用GPU来训练和运行YOLO定制模型。 但是，由于我没有图形处理器，我很困惑该怎么办？因为我买不起这样的图形处理器。此外，我读到关于谷歌可乐，但我不能使用它，我想使用我的模型在离线系统。 在GitHub看到程序中使用的YOLO的系统使用率后，我很害怕： https://github.com/AhmadYahya97/Fully-Automated-red-light-Violation-Detection.git。 我在我的笔记本电脑上运行这个配置： 内存：4 GB 处理器：英特尔i3，2.40 GHz 操作系统：Ubuntu 18.04 LTS

推荐答案

虽然速度会慢很多，但你只能在训练和预测时使用。如果您使用的是原始的DarkNet框架，请在安装DarkNet时将Makefile中的GPU标志设置为GPU=0。

如何安装暗网：https://pjreddie.com/darknet/install/

然后您可以按照以下指导开始训练或预测：https://pjreddie.com/darknet/yolo/

这篇关于我们可以用CPU代替GPU来训练目标检测的YOLO模型吗的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！