opencv 我们可以使用CPU代替GPU来训练定制的YOLO模型进行目标检测吗

5jvtdoz2  于 2023-02-16  发布在  其他
关注(0)|答案(1)|浏览(196)

我想训练我的自定义对象数据集的YOLO模型。我读到它无处不在的各种网站和每个人都在谈论GPU应该用来训练和运行YOLO自定义模型。但是,由于我没有GPU我不知道该怎么做?因为我不能买GPU的。此外,我读到谷歌Colab,但我不能使用它,我想在离线系统上使用我的模型,看到github程序中使用的YOLO的系统利用率后我很害怕:https://github.com/AhmadYahya97/Fully-Automated-red-light-Violation-Detection.git。我在我的笔记本电脑上运行此配置:RAM:4 GB处理器:英特尔i3,2.40 GHz操作系统:Ubuntu 18.04语言

vcirk6k6

vcirk6k61#

虽然它会慢很多,是的,你可以只在训练和预测时使用CPU。如果你使用的是原始的Darknet框架,在安装Darknet到GPU=0时在Makefile中设置GPU标志。
如何安装暗网:https://pjreddie.com/darknet/install/
然后,你就可以开始按照以下指南进行训练或预测:https://pjreddie.com/darknet/yolo/

相关问题