Feature: store npz file from feature feature activations

mseitzer / pytorch-fid

Compute FID scores with PyTorch.

Apache License 2.0

3.22k stars 497 forks source link

Feature: store npz file from feature feature activations #77

Closed ghost closed 5 months ago

ghost commented 2 years ago

Hi, how to use pytorch-fid to produce pre-calculated npz file?

mseitzer commented 2 years ago

This is currently not possible, but I agree it should be implemented.

jwblangley commented 2 years ago

Hi @liuluping0106 ! I have implemented this change and have submitted a PR: #80. Whilst we wait for @mseitzer to take a look, feel free to check out my fork and branch and use the code!