RS-ESRGAN训练后模块导入失败求助(Google Colab环境)
解决RS-ESRGAN自定义模块导入的黄色下划线问题
为啥会出现这问题
options.options、models这些模块是该GitHub仓库内部自定义的模块,并非PyPI上可通过pip安装的公开包,所以用pip search肯定搜不到。Colab编辑器标黄色下划线,本质是编辑器无法识别这些模块的存放路径,属于Python模块搜索路径配置问题。
具体解决办法
1. 完整克隆仓库,保留目录结构
别只单独下载train_esrgan_WV.py文件,在Colab执行以下命令完整拉取仓库,确保options/、models/等模块文件夹和训练脚本处于同一根目录:
!git clone https://github.com/luissalgueiro/rs-esrgan.git %cd rs-esrgan
2. 将仓库根目录加入Python搜索路径
如果克隆后仍有黄色下划线,在train_esrgan_WV.py开头添加以下代码,手动告诉Python去哪里找自定义模块:
import sys from pathlib import Path # 获取当前脚本所在的仓库根目录 repo_root = Path(__file__).parent sys.path.append(str(repo_root))
或者直接在Colab代码单元格中先执行:
import sys sys.path.append('/content/rs-esrgan') # Colab中仓库默认路径,可根据实际情况调整
3. 验证模块导入是否正常
新建Colab代码单元格,测试导入命令,确认路径配置有效:
from options.options import Options from models import esrgan
成功导入后,Colab编辑器的黄色下划线提示会自动消失。
4. 安装仓库全部依赖
先安装仓库requirements.txt中列出的所有基础依赖,避免因依赖缺失导致间接导入问题:
!pip install -r requirements.txt
补充说明
Colab的黄色下划线有时只是编辑器的语法提示警告,不影响代码运行,但按上述步骤配置路径后,既能消除提示,也能确保模块导入的稳定性。
内容的提问来源于stack exchange,提问作者cemre aldoğan
相关产品推荐
相关产品推荐

