我们可以在没有图形处理器的情况下训练YOLOv3自定义对象吗？ [英] Can we Train YOLOv3 on custom objects with no GPU?

问题描述

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

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

推荐答案

是，您可以这样做。

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

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

更改为-

GPU=0 CUDNN=0(用于CPU)

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

这篇关于我们可以在没有图形处理器的情况下训练YOLOv3自定义对象吗？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！