load the frame interpolation model.
(ckpt)
| 91 | |
| 92 | |
| 93 | def load_model(ckpt): |
| 94 | """ |
| 95 | load the frame interpolation model. |
| 96 | """ |
| 97 | params = network_cfg.get("params", {}) |
| 98 | model = Model(**params) |
| 99 | model.load_state_dict(ckpt["state_dict"]) |
| 100 | model = model.to(device) |
| 101 | model.eval() |
| 102 | return model |
| 103 | |
| 104 | |
| 105 | def interpolater(model, inputs, scale, padder, iters=1): |
no test coverage detected