You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.18 09:34:57