Update README.md
Browse files
README.md
CHANGED
|
@@ -8,22 +8,22 @@ tags:
|
|
| 8 |
---
|
| 9 |
# Soft Query Knowledge Distillation
|
| 10 |
This model provides the main distillation experiment results on dataset ScannetV2 and S3DIS for the 3D instance segmentation task in the model zoo, along with the inference code for the [Mask3D](https://github.com/JonasSchult/Mask3D) model and the [SPFormer3D](https://github.com/sunjiahao1999/SPFormer) method. The table below summarizes the main distillation results from our experiments. You can click on ckpt to download the corresponding Baseline model and the distilled model for each experimental group.
|
| 11 |
-
| ID | Baseline mAP / mAP50 / mAP25| SQKD mAP | SQKD mAP50 | SQKD mAP25 | download |
|
| 12 |
-
|-----|-----------------------------|----------|------------|------------|----------|
|
| 13 |
-
| 1 | 24.6 / 44.5 / 65.4 | 30.6 | 52.0 | 70.6 | [ckpt1](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/01.zip)|
|
| 14 |
-
| 2 | 37.7 / 63.0 / 79.0 | 41.0 | 66.3 | 81.2 | [ckpt2](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/02.zip)|
|
| 15 |
-
| 3 | 29.7 / 48.8 / 66.5 | 32.2 | 53.8 | 71.1 | [ckpt3](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/03.zip)|
|
| 16 |
-
| 4 | 43.0 / 68.2 / 81.9 | 46.0 | 71.4 | 84.0 | [ckpt4](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/04.zip)|
|
| 17 |
-
| 5 | 42.2 / 68.9 / 83.1 | 47.2 | 73.7 | 86.4 | [ckpt5](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/05.zip)|
|
| 18 |
-
| 6 | 42.4 / 68.0 / 82.3 | 47.6 | 72.6 | 84.1 | [ckpt6](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/06.zip)|
|
| 19 |
-
| 7 | 36.8 / 54.0 / 66.6 | 41.1 | 58.6 | 71.1 | [ckpt7](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/07.zip)|
|
| 20 |
-
| 8 | 38.3 / 56.3 / 68.4 | 41.7 | 59.6 | 70.8 | [ckpt8](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/08.zip)|
|
| 21 |
-
| 9 | 37.2 / 54.9 / 66.6 | 40.1 | 57.6 | 68.0 | [ckpt9](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/09.zip)|
|
| 22 |
-
| 10 | 25.1 / 40.6 / 53.1 | 30.3 | 45.8 | 57.7 | [ckpt10](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/10.zip)|
|
| 23 |
-
| 11 | 19.9 / 27.1 / 39.4 | 23.2 | 30.8 | 42.0 | [ckpt11](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/11.zip)|
|
| 24 |
-
| 12 | 24.5 / 37.8 / 52.4 | 30.8 | 43.3 | 56.9 | [ckpt12](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/12.zip)|
|
| 25 |
-
| 13 | 22.0 / 30.8 / 45.4 | 27.4 | 36.4 | 47.6 | [ckpt13](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/13.zip)|
|
| 26 |
-
| 14 | 33.3 / 44.3 / 55.3 | 38.9 | 49.9 | 59.4 | [ckpt14](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/14.zip)|
|
| 27 |
|
| 28 |
In the table, experimental groups 1–6, 7–10, and 11–14 use different [teacher models](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/teachers.zip), respectively. We also fixed some bugs in the label category code during data preprocessing and provide the improved versions of the [Mask3D](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/code_mix3d.zip) and [SPFormer](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/code_SPFormer.zip) models.
|
| 29 |
|
|
|
|
| 8 |
---
|
| 9 |
# Soft Query Knowledge Distillation
|
| 10 |
This model provides the main distillation experiment results on dataset ScannetV2 and S3DIS for the 3D instance segmentation task in the model zoo, along with the inference code for the [Mask3D](https://github.com/JonasSchult/Mask3D) model and the [SPFormer3D](https://github.com/sunjiahao1999/SPFormer) method. The table below summarizes the main distillation results from our experiments. You can click on ckpt to download the corresponding Baseline model and the distilled model for each experimental group.
|
| 11 |
+
| ID | Baseline mAP / mAP50 / mAP25|Comp.Ratio| Flops | SQKD mAP | SQKD mAP50 | SQKD mAP25 | download |
|
| 12 |
+
|-----|-----------------------------|----------|----------|----------|------------|------------|----------|
|
| 13 |
+
| 1 | 24.6 / 44.5 / 65.4 | *4.17 | 83% | 30.6 | 52.0 | 70.6 | [ckpt1](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/01.zip)|
|
| 14 |
+
| 2 | 37.7 / 63.0 / 79.0 | *4.17 | 83% | 41.0 | 66.3 | 81.2 | [ckpt2](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/02.zip)|
|
| 15 |
+
| 3 | 29.7 / 48.8 / 66.5 | *2.32 | 95% | 32.2 | 53.8 | 71.1 | [ckpt3](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/03.zip)|
|
| 16 |
+
| 4 | 43.0 / 68.2 / 81.9 | *2.32 | 95% | 46.0 | 71.4 | 84.0 | [ckpt4](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/04.zip)|
|
| 17 |
+
| 5 | 42.2 / 68.9 / 83.1 | *2.32 | 76% | 47.2 | 73.7 | 86.4 | [ckpt5](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/05.zip)|
|
| 18 |
+
| 6 | 42.4 / 68.0 / 82.3 | *2.32 | 57% | 47.6 | 72.6 | 84.1 | [ckpt6](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/06.zip)|
|
| 19 |
+
| 7 | 36.8 / 54.0 / 66.6 | *2.44 | 91% | 41.1 | 58.6 | 71.1 | [ckpt7](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/07.zip)|
|
| 20 |
+
| 8 | 38.3 / 56.3 / 68.4 | *2.44 | 91% | 41.7 | 59.6 | 70.8 | [ckpt8](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/08.zip)|
|
| 21 |
+
| 9 | 37.2 / 54.9 / 66.6 | *2.44 | 71% | 40.1 | 57.6 | 68.0 | [ckpt9](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/09.zip)|
|
| 22 |
+
| 10 | 25.1 / 40.6 / 53.1 | *3.87 | 80% | 30.3 | 45.8 | 57.7 | [ckpt10](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/10.zip)|
|
| 23 |
+
| 11 | 19.9 / 27.1 / 39.4 | *4.17 | 83% | 23.2 | 30.8 | 42.0 | [ckpt11](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/11.zip)|
|
| 24 |
+
| 12 | 24.5 / 37.8 / 52.4 | *4.17 | 83% | 30.8 | 43.3 | 56.9 | [ckpt12](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/12.zip)|
|
| 25 |
+
| 13 | 22.0 / 30.8 / 45.4 | *2.32 | 95% | 27.4 | 36.4 | 47.6 | [ckpt13](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/13.zip)|
|
| 26 |
+
| 14 | 33.3 / 44.3 / 55.3 | *2.32 | 95% | 38.9 | 49.9 | 59.4 | [ckpt14](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/14.zip)|
|
| 27 |
|
| 28 |
In the table, experimental groups 1–6, 7–10, and 11–14 use different [teacher models](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/Compress_Result/teachers.zip), respectively. We also fixed some bugs in the label category code during data preprocessing and provide the improved versions of the [Mask3D](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/code_mix3d.zip) and [SPFormer](https://huggingface.co/Mo-nan/SQKD-model-zoo/resolve/main/code_SPFormer.zip) models.
|
| 29 |
|