StyleGAN源码之FID

看到很多网友在用StyleGAN训练自己的数据集的时候都忽略了fid,源码中fid利用了inception v3的pertained model,源码写的是要从谷歌网盘下载该pretrained model,如果只想run着玩可以忽略fid的计算,把相应代码注释掉;否则最好提前下载下来然后在本地目录导入下载好的inception v3即可。
源码FID的位置:
metrics/frechet_inception_distance.py
调用FID的位置:
metrics/metric_base.py
实际的调用过程为:
MetricBase的run(metric_base.py)–>_evaluate(frechet_inception_distance.py对MetricBase的_evaluate进行了重写)
其中inception v3的pertained model的load代码也在frechet_inception_distance.py中。
如果你只想run着玩,可以去training/training_loop.py文件中把MetricGroup类实例化的对象metrics的相应方法调用注释掉。

你可能感兴趣的:(深度学习,计算机视觉,stylegan,stylegan的fid)