()
| 136 | train_feature_matching(args, model, feat_model, optimizer, i_split, hwf, near, far, device, early_stopping, train_dl=train_dl, val_dl=val_dl, test_dl=test_dl) |
| 137 | |
| 138 | def eval(): |
| 139 | print(parser.format_values()) |
| 140 | # Load data |
| 141 | if args.dataset_type == '7Scenes': |
| 142 | train_dl, val_dl, test_dl, hwf, i_split, near, far = load_7Scenes_dataloader(args) |
| 143 | elif args.dataset_type == 'Cambridge': |
| 144 | train_dl, val_dl, test_dl, hwf, i_split, near, far = load_Cambridge_dataloader(args) |
| 145 | else: |
| 146 | print("please choose dataset_type: 7Scenes or Cambridge, exiting...") |
| 147 | sys.exit() |
| 148 | |
| 149 | # load pretrained DFNet_dm model |
| 150 | model = load_exisiting_model(args) |
| 151 | if args.freezeBN: |
| 152 | model = freeze_bn_layer(model) |
| 153 | model.to(device) |
| 154 | |
| 155 | print(len(test_dl.dataset)) |
| 156 | |
| 157 | get_error_in_q(args, test_dl, model, len(test_dl.dataset), device, batch_size=1) |
| 158 | |
| 159 | if __name__ == '__main__': |
| 160 | if args.eval: |
no test coverage detected