【代码实践】HAT代码Window平台下运行实践记录

一的千分之一

已于 2023-10-31 17:39:17 修改

阅读量1k

点赞数 3

分类专栏：【代码实践】文章标签： python 深度学习

于 2023-10-07 23:01:36 首次发布

本文链接：https://blog.csdn.net/yideqianfenzhiyi/article/details/133659622

版权

【代码实践】专栏收录该内容

2 篇文章 0 订阅

订阅专栏

HAT是CVPR2023上的自然图像超分辨率重建论文《activating More Pixels in Image Super-Resolution Transformer》所提出的模型。本文旨在记录在Window系统下运行该官方代码（https://github.com/XPixelGroup/HAT）的过程，中间会遇到一些问题，供大家参考。

环境安装

参考官方代码，进行环境安装

pip install -r requirements.txt
python setup.py develop

代码运行

这里我在遥感图像的数据集UCMerced上进行重新训练，需要首先在./options/train文件夹中设置对应的yaml文件，然后执行训练过程：

python hat/train.py -opt options/train/train_HAT_SRx4_UCMerced_from_scratch.yml

开始训练了
在这里插入图片描述

遇到的问题

问题1: UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0xae

由于相关开源代码都是在linux下编写和调试的，放在window系统中会出现不少问题，如下读取yaml文件时遇到的问题：
在这里插入图片描述
使用notepad++ “编码”查看对应文件的编码格式。修改报错位置代码，使用‘utf-8’编码格式：

with open(filename, 'r+', encoding='utf-8') as f:

《关于UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte的解决办法》
《真正解决Windows下UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0xff in position 0错误的方法》

问题2: TypeError: expected str, bytes or os.PathLike object, not NoneType

因为自己在训练配置文件yml中，没有设置meta_info的信息，所以报下面的问题：
在这里插入图片描述
参考hat/data/meta_info/meta_info_DF2Ksub_GT.txt:

解决方式有两种：一种是将yml中meta_info给注释掉，或者使用在github/basicsr库中使用generate_meta_info.py文件，生成meta_info文件。我这里直接选择注释掉了。