search

photosynthesis-team / piq

Measures and metrics for image2image tasks. PyTorch.
Apache License 2.0
1.38k stars 116 forks source link

BaseFeatureMetric.compute_feats() warning #307

Open markdjwilliams opened 2 years ago

markdjwilliams commented 2 years ago

Describe the bug

The warning reported by BaseFeatureMetric.compute_feats is likely incorrect:

            print('WARNING: default feature extractor (InceptionNet V2) is used.')
            feature_extractor = InceptionV3()

https://github.com/photosynthesis-team/piq/blob/53c6b8511a54015067d58d669a242574e8da5077/piq/base.py

To Reproduce

Create an instance of the FID class and call compute_feats(....) to see the warning.

Expected behavior The warning should report that InceptionV3 is the default not Inception V2, and ideally use the warnings module to do so instead of print so that applications could filter it out.