(args)
| 150 | |
| 151 | |
| 152 | def load_model_and_dataset(args): |
| 153 | # Read the config file # |
| 154 | with open(args.config_path, 'r') as file: |
| 155 | try: |
| 156 | config = yaml.safe_load(file) |
| 157 | except yaml.YAMLError as exc: |
| 158 | print(exc) |
| 159 | print(config) |
| 160 | ######################## |
| 161 | |
| 162 | dataset_config = config['dataset_params'] |
| 163 | model_config = config['model_params'] |
| 164 | train_config = config['train_params'] |
| 165 | |
| 166 | seed = train_config['seed'] |
| 167 | torch.manual_seed(seed) |
| 168 | np.random.seed(seed) |
| 169 | random.seed(seed) |
| 170 | if device == 'cuda': |
| 171 | torch.cuda.manual_seed_all(seed) |
| 172 | |
| 173 | voc = VOCDataset('test', im_dir=dataset_config['im_test_path'], ann_dir=dataset_config['ann_test_path']) |
| 174 | test_dataset = DataLoader(voc, batch_size=1, shuffle=False) |
| 175 | |
| 176 | faster_rcnn_model = FasterRCNN(model_config, num_classes=dataset_config['num_classes']) |
| 177 | faster_rcnn_model.eval() |
| 178 | faster_rcnn_model.to(device) |
| 179 | faster_rcnn_model.load_state_dict(torch.load(os.path.join(train_config['task_name'], |
| 180 | train_config['ckpt_name']), |
| 181 | map_location=device)) |
| 182 | return faster_rcnn_model, voc, test_dataset |
| 183 | |
| 184 | |
| 185 | def infer(args): |
no test coverage detected