使用Lightning Flash实例化SemanticSegmentationData时出现K未定义错误

问题原因 #

该错误是因为Lightning Flash图像分割模块内部未正确导入Kornia库的别名K，通常由依赖安装不完整或版本不匹配导致。

解决步骤 #

• 卸载现有可能存在问题的依赖包：

!pip uninstall -y lightning-flash kornia

• 安装包含图像模块的完整Lightning Flash包，该命令会自动安装兼容的Kornia等依赖：

!pip install 'lightning-flash[image]'

• 若上述步骤未解决问题，尝试安装指定版本的Kornia（与当前Flash版本兼容）：

!pip install kornia==0.6.0

• 也可尝试升级Lightning Flash到最新稳定版：

!pip install lightning-flash --upgrade

验证 #

重新运行数据模块实例化代码，确认错误是否消失：

datamodule = SemanticSegmentationData.from_folders(
    train_folder=x_train_dir,
    train_target_folder=y_train_dir,
    val_folder=x_valid_dir,
    val_target_folder=y_valid_dir,
    test_folder=x_test_dir,
    test_target_folder=y_test_dir,
    transform_kwargs=dict(image_size=(256, 256)),
    num_classes=1,
    batch_size=16,
)

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