enumerate(train_loader)时报错:AttributeError: Caught AttributeError in DataLoader worker process 0.

标题给出的这行报错信息,没有任何参考价值!应该查看报错信息的其它部分!

对于我遇到的问题:自定义dataset中的图片路径是png,但实际数据集中的图片路径是jpg,将它们处理一下就好了

报错如下

--- load dataset ---
The size of dataset:  10
dataset size: 10
[ WARN:0@3.305] global loadsave.cpp:244 cv::findDecoder imread_('MT_data/seg1\non_makeup\00294.png'): can't open/read file: check file path/integrity
Traceback (most recent call last):
  File "c:\Users\impossible\Desktop\Fun_AI\proj-main\dataset.py", line 530, in <module>
    for i, data in enumerate(train_loader):
  File "E:\app\python3\lib\site-packages\torch\utils\data\dataloader.py", line 631, in __next__
    data = self._next_data()
  File "E:\app\python3\lib\site-packages\torch\utils\data\dataloader.py", line 1346, in _next_data
    return self._process_data(data)
  File "E:\app\python3\lib\site-packages\torch\utils\data\dataloader.py", line 1372, in _process_data
    data.reraise()
  File "E:\app\python3\lib\site-packages\torch\_utils.py", line 705, in reraise
    raise exception
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "E:\app\python3\lib\site-packages\torch\utils\data\_utils\worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)  # type: ignore[possibly-undefined]
  File "E:\app\python3\lib\site-packages\torch\utils\data\_utils\fetch.py", line 51, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "E:\app\python3\lib\site-packages\torch\utils\data\_utils\fetch.py", line 51, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "c:\Users\impossible\Desktop\Fun_AI\proj-main\dataset.py", line 244, in __getitem__
    non_makeup_parse = self.load_parse(
  File "c:\Users\impossible\Desktop\Fun_AI\proj-main\dataset.py", line 58, in load_parse
    parse = self.rotate(parse, angle)
  File "c:\Users\impossible\Desktop\Fun_AI\proj-main\dataset.py", line 62, in rotate
    img = Image.fromarray(img)
  File "E:\app\python3\lib\site-packages\PIL\Image.py", line 3266, in fromarray
    arr = obj.__array_interface__
AttributeError: 'NoneType' object has no attribute '__array_interface__'

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值