site-packages/torch/utils/data/_utils/collate.py中报runtimeerror可能是什么原因?

虽然不知道原理,但是看到collate.py或者runtimeerror中提示类型错误,可以去dataset的getitem看看返回数据的数据类型,数据类型需要是tensor,如果不是tensor,那么dataloader可能出现这类错误

Traceback (most recent call last): File "test.py", line 345, in <module> v5_metric=opt.v5_metric File "test.py", line 107, in test for batch_i, (img, targets, paths, shapes) in enumerate(tqdm(dataloader, desc=s)): File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\tqdm\std.py", line 1195, in __iter__ for obj in iterable: File "F:\git\yolov7-main\yolov7-main\utils\datasets.py", line 109, in __iter__ yield next(self.iterator) File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\utils\data\dataloader.py", line 628, in __next__ data = self._next_data() File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\utils\data\dataloader.py", line 1333, in _next_data return self._process_data(data) File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\utils\data\dataloader.py", line 1359, in _process_data data.reraise() File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\_utils.py", line 543, in reraise raise exception RuntimeError: Caught RuntimeError in DataLoader worker process 1. Original Traceback (most recent call last): File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\utils\data\_utils\worker.py", line 302, in _worker_loop data = fetcher.fetch(index) File "C:\Users\Dell\.conda\envs\sparsercnn\lib\site-packages\torch\utils\data\_utils\fetch.py", line 61, in fetch return self.collate_fn(data) File "F:\git\yolov7-main\yolov7-main\utils\datasets.py", line 434, in collate_fn return torch.stack(img, 0), torch.cat(label, 0), path, shapes RuntimeError: stack expects each tensor to be equal size, but got [1539, 448, 672] at entry 0 and [12, 448, 672] at entry 1
05-19
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值