()
| 141 | return image1, image2, viz_fn |
| 142 | |
| 143 | def build_model(): |
| 144 | print(f"building model...") |
| 145 | cfg = get_cfg() |
| 146 | model = torch.nn.DataParallel(build_flowformer(cfg)) |
| 147 | model.load_state_dict(torch.load(cfg.model)) |
| 148 | |
| 149 | model.cuda() |
| 150 | model.eval() |
| 151 | |
| 152 | return model |
| 153 | |
| 154 | def visualize_flow(root_dir, viz_root_dir, model, img_pairs, keep_size): |
| 155 | weights = None |
no test coverage detected