用SSD训练自己的数据集时报错

最新推荐文章于 2021-01-18 22:05:32 发布

苏格兰橘猫

最新推荐文章于 2021-01-18 22:05:32 发布

阅读量964

点赞数 1

分类专栏：深度学习文章标签： python ssd

本文链接：https://blog.csdn.net/qq_34951080/article/details/78341125

版权

在使用SSD进行训练时遇到错误，原因是数据集中部分图片非RGB格式。通过Python脚本批量转换所有图片至RGB格式，并在images文件夹中保存转换后的图像。

摘要由CSDN通过智能技术生成

训练时出现如下错误:

F1023 20:58:25.679929 10640 annotated_data_layer.cpp:205] Check failed: std::equal(top_shape.begin() + 1, top_shape.begin() + 4, shape.begin() + 1)

这是因为数据集里有些图片不是RGB格式，用python脚本把所有图片全部转化:

from PIL import Image
import os
#读取原始文件夹里的图片
path = <

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

苏格兰橘猫

关注关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

查找数据集中非RGB图像

weixin_41564450的博客

12-25

1686

训练数据集时报以下警告原因是数据集中含有RGBA图像，可使用以下代码查找哪些是RGBA图像（通道数为4） from os.path import join from os import listdir, makedirs from PIL import Image DIR = 'F:/computer_vision/dataset/Species/Training_Set/Trigloporus lastoviza' #需改成你的路径 ls = list(listdir(DIR)) for p in l

ssd训练错误

大可的杨先森的博客

09-26

1298

今天，填充数据重新训练SSD时出现错误 OpenCV Error: Assertion failed ((scn == 3 || scn == 4) && (depth == CV_8U || depth == CV_32F)) in cvtColor, file /build/opencv-SviWsf/opencv-2.4.9.1+dfsg/modules/imgproc/src/color

参与评论您还未登录，请先登录后发表或查看评论

sd 错误Check failed: std::equal(top_shape.begin() + 1, top_shape.begin() + 4, shape.begin() + 1) 解决办法

qq_29007291的博客

12-03

487

近期在用自己的数据训练ssd，前期数据准备工作进展十分顺利，本人是想用coco转化成voc数据集对ssd网络进行训练但是在进行网络训练的时候会显示这样的错误： Check failed: std::equal(top_shape.begin() + 1, top_shape.begin() + 4, shape.begin() + 1) 通过研究代码会发现这是在图片进行resize时显示的错误。...

std::accumulate的用法

最新发布

09-04

关于 SSD 报错 KeyError 的问题，这通常是由于代码中的字典键错误引起的。KeyError 表示你尝试访问一个字典中不存在的键。要解决 KeyError，你可以尝试以下几个步骤： 1. 检查代码中的字典键是否正确：确认你正在...

MobileNet SSD 训练自己的数据集

小猫猫的博客

09-07

1万+

记录下mobilnet-ssd如何跑自己的数据集。默认环境已经配置好，并且demo.py已经可以成功运行了。 mobilenet-ssd下载路径（需要注意的点：需要以Git的方式下载，不要以zip的方式下载，否则后面训练时可能报错） $ git clone https://github.com/chuanqi305/MobileNet-SSD.git 步骤记录：1、制作VOC数据集 ...

SSD-Pytorch模型训练自己的数据集

科研鬼才的博客

04-22

7707

开始之前声明: 博主用的环境是ubuntu18.04+anaconda3+pytorch1.4+pycharm 1.下载SSD-Pytorch代码 SSD-pytorch代码链接： https://github.com/amdegroot/ssd.pytorch git clone https://github.com/amdegroot/ssd.pytorch 运行该代码下载到本地（如果下...

成功运行SSD_Pytorch并训练自己的数据集

m0_47452894的博客

01-18

1万+

目录前言一、环境二、准备数据集1.voc2007数据集2.voc2012数据集三、从Github上下载作者源码创建文件夹四、修改代码1.修改data/config.py2.修改data/voc0712.py3.修改layers/modules/multibox_loss.py4.修改train.py5.修改ssd.py五、开始训练六、运行eval.py1.修改eval.py2.修改ssd.py七、运行test.py修改test.py八、运行demo.ipynb修改demo.py总结前言本文主要是记录自己

在训练Pytorch ssd时出现的问题

weixin_42631693的博客

07-16

4094

1.The shape of the mask [32, 8732] at index 0 does not match the shape of the indexed tensor [279424, 1] at index 0 张量大小不匹配,解决办法: 在multibox_loss.py的 98 loss_c[pos] = 0 # filter out pos boxes for ...

SSD-PyTorch训练自己的数据集问题汇总

lianggyu的博客

09-04

5871

代码链接：https://github.com/amdegroot/ssd.pytorch 运行train.py : （1）ModuleNotFoundError: No module named 'cv2' 安装opencv-python ：pip install opencv-python （2）FileNotFoundError: [Errno 2] No such...

使用SSD（caffe）训练自己的数据集以及遇到的问题

qq_41706331的博客

04-19

824

本人刚刚接触深度学习一个月左右，也是第一次使用SSD框架，很多东西可能描述不准确，大家多多包涵！ 1.数据集准备首先，我们需要一个数据集，数据集的标准是按照VOC2007制作的，如何制作我并不是很了解就不说了，我是老师给的数据集，数据集也存在不少坑，后面会讲。 VOC数据集格式：包涵Annotation、ImageSets、JPEGImages...

训练SSD时出现的数据问题

ZL.dreamer的博客

06-28

1631

使用COCO2014训练SSD网络。参照https://github.com/weiliu89/coco.git的方法处理数据集。生成lmdb格式数据。开始训练之后报这种错误。OpenCV Error: Assertion failed ((scn == 3 || scn == 4) && (depth == CV_8U || depth == CV_32F)) in cvtCol...

如何将caffe的data prefetch变为多线程预取？

thankyvision

04-17

1589

在深度学习研究时，数据预取会经常成为训练的瓶颈，一般来说，判断瓶颈的时候，看GPU利用率即可。通过nvidia-smi或者watch -n 1 nvidia-smi（实时刷新）都可查看GPU的利用率。如上图所示，GPU利用率不能完全打满，断断续续的忽高忽低，大概率是因为数据取出速度慢（其他原因，如GPU带宽一般不会成为瓶颈）。除此之外，在blocking_queue.cpp中也会打印出...

深度学习ssd检测模型训练自己的数据集

liu_xiao_cheng的专栏

11-17

4688

各种教学视频或文档资料 +QQ：635992897 之前参考过几篇文章发现没有可以走通的训练自己的数据集的例子，根据网上的几篇文章自己做了个完整的，并且已验证训练出的model paper:https://arxiv.org/abs/1512.02325 作者代码：https://github.com/weiliu89/caffe/tree/ssd 制作VOC数据...