ESRGAN复现时出现问题 AttributeError: 'NoneType' object has no attribute 'astype'

最新推荐文章于 2022-03-11 18:34:50 发布

ztl123452010

最新推荐文章于 2022-03-11 18:34:50 发布

阅读量2.8k

点赞数 1

文章标签： esrgsn srgan AttributeError NoneType astype

本文链接：https://blog.csdn.net/ztl123452010/article/details/98779619

版权

在尝试复现ESRGAN代码时遇到AttributeError，具体为'NoneType' object has no attribute 'astype'。错误源于torch.utils.data.dataloader中的collate_fn，经追踪定位到data.util.py的read_img函数中，可能原因是HR_env变量未正确加载图像导致。寻求解决方案。

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

ESRGAN源代码链接https://github.com/xinntao/BasicSR

我在复现时遇到了以下问题，久久不能解决，望大神指导

以下为报错的详细信息：

Traceback (most recent call last):
File "/sda/ZTL/B/codes/train.py", line 173, in <module>
main()
File "/sda/ZTL/B/codes/train.py", line 97, in main
for _, train_data in enumerate(train_loader):
File "/root/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in __next__
return self._process_next_batch(batch)
File "/root/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
AttributeError: Traceback (most recent cal