(self, use_gpu=True)
| 47 | |
| 48 | class LPIPS(): |
| 49 | def __init__(self, use_gpu=True): |
| 50 | self.model = dm.DistModel() |
| 51 | self.model.initialize(model='net-lin', net='alex', use_gpu=use_gpu) |
| 52 | self.use_gpu = use_gpu |
| 53 | |
| 54 | def __call__(self, image_1, image_2): |
| 55 | """ |
nothing calls this directly
no test coverage detected