Anyone can tell me how to use FId

photosynthesis-team / piq

Measures and metrics for image2image tasks. PyTorch.

Apache License 2.0

1.32k stars 114 forks source link

Anyone can tell me how to use FId #380

Closed LemonC19 closed 3 months ago

LemonC19 commented 3 months ago

import torch from torch.utils.data import DataLoader from piq import FID

first_dl, second_dl = DataLoader(), DataLoader() fid_metric = FID() first_feats = fid_metric.compute_feats(first_dl) second_feats = fid_metric.compute_feats(second_dl) fid: torch.Tensor = fid_metric(first_feats, second_feats)