()
| 51 | |
| 52 | |
| 53 | def init_dm_model(): |
| 54 | global dm_model |
| 55 | if dm_model is not None: |
| 56 | return |
| 57 | |
| 58 | logger.info("Loading PerceptualSimilarity model ..") |
| 59 | dm_model = PerceptualSimilarity.dist_model.DistModel() |
| 60 | dm_model.initialize(model='net-lin', net='alex', use_gpu=True) |
| 61 | |
| 62 | |
| 63 | def parse_args(): |
no test coverage detected