hukkelas/pytorch-frechet-inception-distance

first_image shape is inverted in load_image function

muneeb2001 opened this issue · 2 comments

muneeb2001 commented

ERROR
Traceback (most recent call last):
File "fid.py", line 280, in <module>
images1 = load_images(options.path1)
File "fid.py", line 251, in load_images
final_images[idx] = im
ValueError: could not broadcast input array from shape (128,96,3) into shape (96,128,3)

This occurs because at line 242 W, H = first_image.shape[:2] height and width dimensions are switched.

KomputerMaster64 commented

Thank you for the solution

  • 👍1
hukkelas commented

This repository is deprecated and I recommend you to use torch fidelity instead: https://github.com/toshas/torch-fidelity