我们可以在没有图形处理器的情况下训练YOLOv3自定义对象吗?我们可以、自定义、处理器、情况下

2023-09-03 09:34:39 作者:拔剑斩情丝

我的YOLO模型适用于检测瓶子、人、手机、背包等物体。但我想让我的模型检测到戒指、手镯或头盔(在目前的yolo模型中不存在的对象)。没有GPU,我可以做一个定制的物体检测yolo模型吗?涉及的风险是什么?(如果有)。

我的系统是Windows 10家庭单语版,内存为8 GB。

推荐答案

在ubuntu18.04下训练yolov3模型 GPU

是,您可以这样做。

只需在DarkNet文件夹的Makefile中更改以下行-

GPU=1 CUDNN=1(适用于GPU)

更改为-

GPU=0 CUDNN=0(用于CPU)

是的,您也可以训练您的YOLO模型来检测自定义对象。只需关注本博客-Link