Win10下ImageAI-gpu训练自己的数据集

1.先按《Win10下ImageAI-gpu(tensorflow-gpu)环境安装（cuda及cuDNN自动安装）及实时对象检测》安装好。

2.下载pretrained-yolov3.h5，放到temp.py同一目录下。

3.下载hololens数据集，放到temp.py同一目录下。

4.编写代码开始训练。

from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory="hololens")
trainer.setTrainConfig(object_names_array=["hololens"], batch_size=2, num_experiments=100, train_from_pretrained_model="pretrained-yolov3.h5")
trainer.trainModel()

batch_size=2是关键，否则可能出现以下错误

Resource exhausted: OOM when allocating tensor with shape[8,1024,14,14] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

Win10下ImageAI-gpu训练自己的数据集相关推荐

Win10下用TensorFlow训练自己的数据集来做目标检测
本文主要来源于YouTube博主 EdjeElectronics的视频https://www.youtube.com/watch?v=Rgpfk6eYxJA (对于视频看不了的朋友,我把视频录制下来放 ...
笔记：caffe安装编译，并用ssd gpu训练自己的数据集
caffe ssd GPU训练自己的数据集一.caffe环境搭建二.数据集三.训练和测试一.caffe环境搭建环境搭建过程网上有很多教程,可能要踩些坑吧,正常,这里不多赘述,主要是我当时也没 ...
笔记：caffe ssd gpu训练自己的数据集
caffe ssd GPU训练自己的数据集一.caffe环境搭建二.数据集三.训练和测试一.caffe环境搭建环境搭建过程网上有很多教程,可能要踩些坑吧,正常,这里不多赘述,主要是我当时也没 ...
一步一步教你在 docker 容器下使用 mmdetection 训练自己的数据集
这里不再介绍 mmdetection 的安装和配置,使用 mmdetection 较简单的方法是使用已安装 mmdetection 的 docker 容器.这样直接省去了安装 mmdetection ...
Windows下使用Darknet训练自己的数据集（模型：yolov4-tiny、数据集：垃圾分类）
本文章主要介绍如何使用Darknet在windows下训练自己的数据集,其中模型使用的是yolov4-tiny,数据集使用的是自己垃圾分类数据集(需要的自取:在我上传的资源中有) PS:这是我的第一篇 ...
YOLOv7使用云GPU训练自己的数据集
使用的云GPU是恒源云,帮助文档很详细,后面如果有时间的话,我再写一篇云GPU怎么用,希望能帮助没有显卡想用云GPU的同学.接下来就进入正题吧使用的是云GPU+pycharm专业版(社区版无法远程链 ...
ubuntu18.04下用yolo训练自己的数据集
ubuntu18.04下用yolo训练自己的数据集并进行检测收集数据集并标注数据集的量对于检测的准确度有很高的影响,所以应该尽可能的获取较多的数据集.这里,我从百度图片上获得了92张夏洛特中的动漫 ...
win10下使用caffe训练自己的数据，车牌二分类
caffe编译参考:http://blog.csdn.net/cym1990/article/details/72630584 1.数据格式转换 Caffe采用leveldb或者lmdb的数据格式. ...
Win10下Tensorflow(GPU版)安装趟坑实录，成功了。
这篇文章不指导你具体怎么安装,其他博客大把安装方法,我只告诉你,具体的版本不要看其他人的,每个人用的CUDA可能都不同,越看越乱,直接看运行后系统的提示,我整了一天没有搞定,最后还是按照系统提示搞定的 ...
WIN10下TensorFlow GPU版安装顺序
1.更新驱动 2.先安装Anaconda,创建一个虚拟环境 4.再安装TensorFlow-GPU 5.到虚拟环境所在的目录里寻找build_info.py文件,一般如果全部默认安装的话在: C:\U ...

Win10下ImageAI-gpu训练自己的数据集

Win10下ImageAI-gpu训练自己的数据集相关推荐

最新文章

热门文章