Skip to content

Commit 538a776

Browse files
authored
Update README.md
1 parent 802da39 commit 538a776

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

README.md

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,14 @@ This is a port of the official implementation of [Fréchet Inception Distance](h
44
See [https://github.com/bioinf-jku/TTUR](https://github.com/bioinf-jku/TTUR) for the original implementation using Tensorflow.
55

66
FID is a measure of similarity between two datasets of images.
7-
It was shown to correlate well with human judgement of visual quality and is used most often used to evaluate the quality of samples of Generative Adversarial Networks.
7+
It was shown to correlate well with human judgement of visual quality and is most often used to evaluate the quality of samples of Generative Adversarial Networks.
88
FID is calculated by computing the [Fréchet distance](https://en.wikipedia.org/wiki/Fr%C3%A9chet_distance) between two Gaussians fitted to feature representations of the Inception network.
99

10-
Further insights and an independent evaluation of the FID score can be found in [Are GANs Created Equal? A Large-Scale Study](https://arxiv.org/pdf/1711.10337.pdf)
10+
Further insights and an independent evaluation of the FID score can be found in [Are GANs Created Equal? A Large-Scale Study](https://arxiv.org/abs/1711.10337).
11+
12+
**Note that the official implementation most likely gives slightly different scores**, as it uses pretrained weights from Tensorflow 's Inception, not PyTorch's Inception as used here.
13+
If you report FID scores in your paper, and you want them to be comparable to FID scores reported in other papers, you should use [the official Tensorflow implementation](https://github.com/bioinf-jku/TTUR).
14+
You can still use this version if you want a quick FID estimate without installing Tensorflow.
1115

1216
## Usage
1317

@@ -25,8 +29,6 @@ To compute the FID score between two datasets, where images of each dataset are
2529

2630
To run the evaluation on GPU, use the flag `--gpu N`, where `N` is the index of the GPU to use.
2731

28-
Note that the official implementation might give slightly different scores, as it uses pretrained weights from Tensorflow 's Inception, not PyTorch's Inception as used here.
29-
3032
### Using different layers for feature maps
3133

3234
In difference to the official implementation, you can choose to use a different feature layer of the Inception network instead of the default `pool3` layer.

0 commit comments

Comments
 (0)