()
| 240 | global_step += 1 |
| 241 | |
| 242 | def train(): |
| 243 | |
| 244 | print(parser.format_values()) |
| 245 | |
| 246 | # Load data |
| 247 | if args.dataset_type == '7Scenes': |
| 248 | |
| 249 | train_dl, val_dl, hwf, i_split, bds, render_poses, render_img = load_7Scenes_dataloader_NeRF(args) |
| 250 | near = bds[0] |
| 251 | far = bds[1] |
| 252 | |
| 253 | print('NEAR FAR', near, far) |
| 254 | train_nerf(args, train_dl, val_dl, hwf, i_split, near, far, render_poses, render_img) |
| 255 | return |
| 256 | |
| 257 | elif args.dataset_type == 'Cambridge': |
| 258 | |
| 259 | train_dl, val_dl, hwf, i_split, bds, render_poses, render_img = load_Cambridge_dataloader_NeRF(args) |
| 260 | near = bds[0] |
| 261 | far = bds[1] |
| 262 | |
| 263 | print('NEAR FAR', near, far) |
| 264 | train_nerf(args, train_dl, val_dl, hwf, i_split, near, far, render_poses, render_img) |
| 265 | return |
| 266 | |
| 267 | else: |
| 268 | print('Unknown dataset type', args.dataset_type, 'exiting') |
| 269 | return |
| 270 | |
| 271 | if __name__=='__main__': |
| 272 |
no test coverage detected