前言

yolact训练自己的服装数据集，服装数据集采用labelme进行标注，采用labelme2COCO将生成的饿接送文件转换为COCO格式，随后参照https://blog.csdn.net/weixin_40802676/article/details/115413362进行修改。

仅粗糙记录，方便自己以后进行参考！！！！！！

提示：以下是本篇文章正文内容，下面案例可供参考

一、问题１

１、RuntimeError: CUDA out of memory. Tried to allocate 150.00 MiB (GPU 0; 2.00 GiB total capacity; 1.10 GiB already allocated; 138.10 MiB free; 23.67 MiB cached)
这里多半是因为自己的电脑GPU显存不够，将CUDA改为“－１”，准备采用CPU跑，但依旧报此错误，后来采用云端服务器，问题解决。但出现新的问题，如问题２

二、问题２

1.RuntimeError: Expected a ‘cuda’ device type for generator but found ‘cpu’：

解决办法，将train.py中的Dataloader中第249行代码shuffle=true改为False ，问题解决。

三、云端训练

将整个压缩包.zip传入云端之后，在终端输入解压文件代码（unzip yolact_master.zip. 注意：此处的要在该文件的路径上进行解压

四、云端训练结束时出现的错误

查阅资料发现是全局变量与聚币变量没处理好的问题，故将报错出train.py的第374行前添加了global val_dataset。

运行命令：batch_size自己调节。
python3 train.py --config=yolact_base_config --batch_size=2
采用python train.py --config=yolact_base_config会默认batch_size=8，太大会使得GPU内存溢出。

【Yolact训练自己的数据集（踩坑总结）】相关推荐

新手可直接复现：Duan版本CenterNet在2080Ti上训练自己的数据集——踩坑记录
实验室换了GPU,记录下centernet在新电脑上的配置过程,如果对下面很多不理解先阅读我以前的这篇博客即可初学者复现CornerNet:详细指导零基础在Ubuntu系统运行该代码并完全理解论文思路 ...
Mask R-CNN 训练自己的数据集—踩坑与填坑
作者 | Raini 出品 | 北京图特摩斯科技有限公司(thutmose.cn) tensorflow/keras版(项目地址https://github.com/matterport). 服务器c ...
Yolact训练自己的数据集
Yolact训练自己的数据集 1:提前准备好自己的数据集 2:下载github存储库 3:修改config.py 4:训练 5:测试可能是由于yolact官方更新过其项目代码,所以网上其他人的yol ...
Cityscapse 数据集使用 + 训练 STDC踩坑
官网地址:Cityscapes Dataset – Semantic Understanding of Urban Street Scenes (cityscapes-dataset.com) get ...
soloV2保姆级教程（含环境配置，训练自己的数据集，代码逻辑分析等。能踩得坑都踩了....）更新ing
序言环境:ubuntu18.04 cuda11.1 显卡3050 pytorch1.9 代码链接: GitHub - aim-uofa/AdelaiDet: AdelaiDet is an ope ...
faster rcnn 训练自己的数据集－－－踩坑记录！！！
1．下载代码: git clone https://github.com/jwyang/faster-rcnn.pytorch.git 也可以暴力下载<--直接download压缩包--> ...
用Yolact模型训练自己的数据集
1.首先用labelme标注数据(分割法) 2.yolact训练的数据集为coco数据集,所以要转化数据集,使用从Git下载的labelme项目中的labelme2coco转换就可以这里因为我的标注 ...
【Yolact训练自己的数据从实战到调参】
100%完美运行,期间踩过许多坑,数据集的坑.代码的坑都有.现在做一下总结,也希望后来者能快速上手,避免浪费不必要的时间. 文章目录一.训练自己的数据与预测 1.训练自己的数据train.py 1. ...
分布式深度学习最佳入门（踩坑）指南
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达作者丨Lyon@知乎(已授权) 来源丨https://zhuanla ...

【Yolact训练自己的数据集（踩坑总结）】

yolact训练自己的数据集

前言