KPConv-PyTorch/doc/pretrained_models_guide.md
HuguesTHOMAS cf1f06381e .
2021-08-05 20:11:34 +00:00

49 lines
2.2 KiB
Markdown

## S3DIS Pretrained Models
### Models
We provide pretrained weights for S3DIS dataset. The raw weights come with a parameter file describing the architecture and network hyperparameters. The code can thus load the network automatically.
| Name (link) | KPConv Type | Description | Score |
|:-------------|:-------------:|:-----|:-----:|
| [Light_KPFCNN](https://drive.google.com/file/d/14sz0hdObzsf_exxInXdOIbnUTe0foOOz/view?usp=sharing) | rigid | A network with small `in_radius` for light GPU consumption (~8GB) | 65.4% |
| [Heavy_KPFCNN](https://drive.google.com/file/d/1ySQq3SRBgk2Vt5Bvj-0N7jDPi0QTPZiZ/view?usp=sharing) | rigid | A network with better performances but needing bigger GPU (>18GB). | 66.4% |
| [Deform_KPFCNN](https://drive.google.com/file/d/1ObGr2Srfj0f7Bd3bBbuQzxtjf0ULbpSA/view?usp=sharing) | deform | Deformable convolution network needing big GPU (>20GB). | 67.3% |
| [Deform_Light_KPFCNN](https://drive.google.com/file/d/1gZfv6q6lUT9STFh7Fk4qVa5IVTgwmWIr/view?usp=sharing) | deform | Lighter version of the deformable architecture (~8GB). | 66.7% |
### Instructions
1. Unzip and place the folder in your 'results' folder.
2. In the test script `test_any_model.py`, set the variable `chosen_log` to the path were you placed the folder.
3. Run the test script
python3 test_any_model.py
4. You will see the performance (on the subsampled input clouds) increase as the test goes on.
Confusion on sub clouds
65.08 | 92.11 98.40 81.83 0.00 18.71 55.41 68.65 90.93 79.79 74.83 65.31 63.41 56.62
5. After a few minutes, the script will reproject the results form the subsampled input clouds to the real data and get you the real score
Reproject Vote #9
Done in 2.6 s
Confusion on full clouds
Done in 2.1 s
--------------------------------------------------------------------------------------
65.38 | 92.62 98.39 81.77 0.00 18.87 57.80 67.93 91.52 80.27 74.24 66.14 64.01 56.42
--------------------------------------------------------------------------------------
6. The test script creates a folder `test/name-of-your-log`, where it saves the predictions, potentials, and probabilities per class. You can load them with CloudCompare for visualization.