Google Colab计算FID时出现“虚部”错误的技术求助

核心排查方向 #

这类错误通常源于文件读取不完整、环境数值精度差异或Drive挂载的IO问题，以下是具体解决步骤：

1. 验证图片文件完整性 #

首先确认两个文件夹内的图片无损坏、格式一致，避免因文件问题导致计算异常：

import os
from PIL import Image

def validate_images(folder_path):
    total = 0
    corrupted = []
    for filename in os.listdir(folder_path):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
            total += 1
            try:
                with Image.open(os.path.join(folder_path, filename)) as img:
                    img.verify()  # 验证图片完整性
            except Exception as e:
                corrupted.append(filename)
    print(f"文件夹 {folder_path}：有效图片{total}张，损坏图片{len(corrupted)}张")
    return corrupted

# 检查两个目标文件夹
validate_images("/content/drive/MyDrive/pathA")
validate_images("/content/drive/MyDrive/cyclegan/pathB")

• 若存在损坏图片，删除或替换；同时确认图片未被Google Drive自动转码（如转为WebP格式，可右键查看文件属性确认）。

2. 复制文件到Colab本地磁盘计算 #

Drive挂载的远程IO可能导致读取不完整，将文件复制到本地后再计算：

# 复制文件夹到本地
!cp -r "/content/drive/MyDrive/pathA" "/content/local_pathA"
!cp -r "/content/drive/MyDrive/cyclegan/pathB" "/content/local_pathB"

from cleanfid import fid
# 基于本地文件计算FID
score = fid.compute_fid("/content/local_pathA", "/content/local_pathB")
print(f"FID分数：{score}")

3. 强制指定数值精度 #

Colab默认的混合精度计算可能引发协方差矩阵数值不稳定，强制使用float32精度：

from cleanfid import fid
score = fid.compute_fid(
    "/content/local_pathA", 
    "/content/local_pathB",
    dtype="float32",  # 强制单精度计算
    device="cpu"  # 可选：用CPU计算避免GPU精度差异
)
print(f"FID分数：{score}")

若使用pytorch-fid，同样指定精度：

!python -m pytorch_fid "/content/local_pathA" "/content/local_pathB" --dtype float32

4. 对齐依赖版本 #

对比Kaggle与Colab的PyTorch、torchvision、clean-fid版本，确保环境一致：

import torch, torchvision, cleanfid
print(f"PyTorch版本：{torch.__version__}")
print(f"TorchVision版本：{torchvision.__version__}")
print(f"Clean-FID版本：{cleanfid.__version__}")

若版本差异较大，安装与Kaggle一致的版本，例如：

!pip install torch==2.0.1 torchvision==0.15.2 clean-fid==0.1.35

5. 测试最小复现案例 #

创建两个完全相同的文件夹进行自对比，验证环境是否正常：

!cp -r "/content/local_pathA" "/content/local_pathC"

score = fid.compute_fid("/content/local_pathA", "/content/local_pathC")
print(f"自对比FID分数：{score}")

• 正常情况下分数应接近0；若仍报错，重启Colab运行时后重新安装依赖再试。

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