sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
160 stars 25 forks source link

IS运行指令 #73

Closed tomato213 closed 1 month ago

tomato213 commented 1 month ago

作者您好,可以提供一下计算生成的数据的 Inception Score运行指令吗?

sjtuplayer commented 1 month ago

你可以参考https://github.com/toshas/torch-fidelity 安装好torch-fidelity后,运行: fidelity --gpu 0 --isc --input1 dataset_path