Flask项目导入DeOldify模块时出现依赖导入错误的求助

问题根源 #

当从app.py导入DeOldify/image_colorizer.py时，Python解释器的搜索路径（sys.path）不包含DeOldify目录，导致image_colorizer.py无法找到同级的deoldify、fastai等子包。同时代码中存在导入名称与文件名不匹配的潜在问题。

方案1：将DeOldify转为Python包并使用相对导入（推荐） #

步骤1：添加__init__.py文件 #

在revitalize/DeOldify目录下创建__init__.py文件，内容如下：

from .image_colorizer import ColorIMG

这会将DeOldify标记为Python包，并直接导出ColorIMG函数。

步骤2：修改image_colorizer.py的导入方式 #

将image_colorizer.py中的绝对导入改为相对导入，确保它能找到同级的子包：

from .deoldify import device
from .deoldify.device_id import DeviceId
from .deoldify.visualize import *

步骤3：修正app.py的导入与调用 #

将原来的导入语句替换为：

from DeOldify import ColorIMG

然后在colorize路由中直接调用函数：

@app.route('/colorize', methods=['GET','POST'])
def colorize():
    ColorIMG()
    return render_template("index.html")

方案2：手动将DeOldify目录添加到Python路径 #

如果不想修改image_colorizer.py的导入逻辑，可以在app.py中先将DeOldify目录加入Python搜索路径：

步骤1：修改app.py开头代码 #

from flask import Flask, request, render_template, send_from_directory,redirect
from werkzeug.utils import secure_filename
import os
import sys

# 获取DeOldify目录的绝对路径
deoldify_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'DeOldify')
# 将路径添加到sys.path
sys.path.append(deoldify_dir)

# 直接导入image_colorizer模块
import image_colorizer

步骤2：修正路由中的调用 #

@app.route('/colorize', methods=['GET','POST'])
def colorize():
    image_colorizer.ColorIMG()
    return render_template("index.html")

额外注意事项 #

• 检查文件名匹配：确保app.py中的导入名称与实际文件名一致（当前代码中imageColorizer对应image_colorizer.py，若无__init__.py映射会直接报错）。
• 确认子包完整性：deoldify和fastai子目录中需存在__init__.py文件（DeOldify源码通常自带，若缺失需补充）。

内容的提问来源于stack exchange，提问作者Muthu Raj