首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >[深度学习][原创]in function ‘copyMakeBorder‘错误解决方法

[深度学习][原创]in function ‘copyMakeBorder‘错误解决方法

作者头像
云未归来
发布2025-07-18 16:52:38
发布2025-07-18 16:52:38
710
举报

 pytorch yolov3训练或者剪枝时候会提示下面错误

File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 435, in __next__     data = self._next_data()   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1085, in _next_data     return self._process_data(data)   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1111, in _process_data     data.reraise()   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 428, in reraise     raise self.exc_type(msg) cv2.error: Caught error in DataLoader worker process 7. Original Traceback (most recent call last):   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop     data = fetcher.fetch(index)   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch     data = [self.dataset[idx] for idx in possibly_batched_index]   File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>     data = [self.dataset[idx] for idx in possibly_batched_index]   File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210411/utils/datasets.py", line 426, in __getitem__     img, ratio, padw, padh = letterbox(img, self.batch_shapes[self.batch[index]], mode='rect')   File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210411/utils/datasets.py", line 643, in letterbox     img = cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color)  # add border cv2.error: OpenCV(4.5.1) /tmp/pip-req-build-ms668fyv/opencv/modules/core/src/copy.cpp:1445: error: (-215:Assertion failed) top >= 0 && bottom >= 0 && left >= 0 && right >= 0 && _src.dims() <= 2 in function 'copyMakeBorder' 这是因为前面缓存的shapes造成的,删除data文件夹下面train.shapes和test.shapes即可解决问题

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021-04-12,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档