| { | |
| "library_name": "torch-pointcloud", | |
| "name": "pointnext-sm.scanobjectnn-hardest.openpoints", | |
| "task": "classification", | |
| "hparams": { | |
| "in_channels": 4, | |
| "num_classes": 15, | |
| "spatial_dim": 3, | |
| "stem_channels": 32, | |
| "stem_plain_last": true, | |
| "encoder_channels": [ | |
| 64, | |
| 128, | |
| 256, | |
| 512 | |
| ], | |
| "encoder_depths": [ | |
| 0, | |
| 0, | |
| 0, | |
| 0 | |
| ], | |
| "encoder_expansion": 4, | |
| "sa_layers": 2, | |
| "sa_use_res": true, | |
| "ratios": [ | |
| 0.5, | |
| 0.5, | |
| 0.5, | |
| 0.5 | |
| ], | |
| "radiuses": [ | |
| 0.15, | |
| 0.225, | |
| 0.3375, | |
| 0.50625, | |
| 0.759375 | |
| ], | |
| "num_neighbors": [ | |
| 32, | |
| 32, | |
| 32, | |
| 32, | |
| 32 | |
| ], | |
| "act": "relu", | |
| "act_first": false, | |
| "norm": "batch_norm", | |
| "bias": true, | |
| "add_self_loops": false, | |
| "dropout": 0.5, | |
| "head_channels": [ | |
| 512, | |
| 256 | |
| ], | |
| "global_sa_channels": [ | |
| 512, | |
| 512 | |
| ] | |
| }, | |
| "input_keys": [ | |
| "x", | |
| "pos", | |
| "batch" | |
| ], | |
| "dataset": "scanobjectnn-hardest", | |
| "metrics": { | |
| "OA": 88.17, | |
| "mAcc": 86.8 | |
| }, | |
| "classes": [ | |
| "bag", | |
| "bin", | |
| "box", | |
| "cabinet", | |
| "chair", | |
| "desk", | |
| "display", | |
| "door", | |
| "shelf", | |
| "table", | |
| "bed", | |
| "pillow", | |
| "sink", | |
| "sofa", | |
| "toilet" | |
| ], | |
| "author": "openpoints", | |
| "license": "MIT" | |
| } | |