ValueError: Caught ValueError in DataLoader worker process 3.
Original Traceback (most recent call last):
File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\_utils\worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\aaa\envs\pao\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\xiangmu\yolov5.0\utils\datasets.py", line 525, in __getitem__
img, labels = load_mosaic(self, index)
File "D:\xiangmu\yolov5.0\utils\datasets.py", line 698, in load_mosaic
img4[y1a:y2a, x1a:x2a] = img[y1b:y2b, x1b:x2b] # img4[ymin:ymax, xmin:xmax]
ValueError: could not broadcast input array from shape (579,527,3) into shape (579,527,4)
yolov5 自己数据集训练报错
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
1条回答 默认 最新
- 敲蚂蚁的小代码 2022-05-14 17:11关注
赋值时维度不一致,前面的img4是4通道的,后面img是3通道的。
猜测源码这里初始化img4时,训练集中有4通道带深度的图片,可以排查下数据集尝试下。
解决 无用评论 打赏 举报
悬赏问题
- ¥15 执行 virtuoso 命令后,界面没有,cadence 启动不起来
- ¥50 comfyui下连接animatediff节点生成视频质量非常差的原因
- ¥20 有关区间dp的问题求解
- ¥15 多电路系统共用电源的串扰问题
- ¥15 slam rangenet++配置
- ¥15 有没有研究水声通信方面的帮我改俩matlab代码
- ¥15 ubuntu子系统密码忘记
- ¥15 保护模式-系统加载-段寄存器
- ¥15 电脑桌面设定一个区域禁止鼠标操作
- ¥15 求NPF226060磁芯的详细资料