(dataroot, npoints,category,use_mask=False)
| 379 | ############################################################################# |
| 380 | |
| 381 | def get_dataset(dataroot, npoints,category,use_mask=False): |
| 382 | tr_dataset = ShapeNet15kPointClouds(root_dir=dataroot, |
| 383 | categories=[category], split='train', |
| 384 | tr_sample_size=npoints, |
| 385 | te_sample_size=npoints, |
| 386 | scale=1., |
| 387 | normalize_per_shape=False, |
| 388 | normalize_std_per_axis=False, |
| 389 | random_subsample=True, use_mask = use_mask) |
| 390 | te_dataset = ShapeNet15kPointClouds(root_dir=dataroot, |
| 391 | categories=[category], split='val', |
| 392 | tr_sample_size=npoints, |
| 393 | te_sample_size=npoints, |
| 394 | scale=1., |
| 395 | normalize_per_shape=False, |
| 396 | normalize_std_per_axis=False, |
| 397 | all_points_mean=tr_dataset.all_points_mean, |
| 398 | all_points_std=tr_dataset.all_points_std, |
| 399 | use_mask=use_mask |
| 400 | ) |
| 401 | return tr_dataset, te_dataset |
| 402 | |
| 403 | |
| 404 |
no test coverage detected