我想训练我的自定义对象数据集的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语言
1条答案
按热度按时间vcirk6k61#
虽然它会慢很多,是的,你可以只在训练和预测时使用CPU。如果你使用的是原始的Darknet框架,在安装Darknet到
GPU=0
时在Makefile中设置GPU标志。如何安装暗网:https://pjreddie.com/darknet/install/
然后,你就可以开始按照以下指南进行训练或预测:https://pjreddie.com/darknet/yolo/