richzhang/PerceptualSimilarity

LPIPS for other models

MalharJ opened this issue 3 years ago · 1 comments

MalharJ commented 3 years ago

How can I experiment with resnet models? The library currently supports only vgg, alexnet and squeezenet models.

richzhang commented 3 years ago

You can trace from the LPIPS class on down: https://github.com/richzhang/PerceptualSimilarity/blob/master/lpips/lpips.py#L84. It'll involve adding the network to this file: https://github.com/richzhang/PerceptualSimilarity/blob/master/lpips/pretrained_networks.py.

Good luck -- looking forward to seeing what you get