Update task checkpoint and evaluation summary
Browse files
README.md
CHANGED
|
@@ -14,17 +14,25 @@ pipeline_tag: text-generation
|
|
| 14 |
|
| 15 |
# Multi-3DLLM Checkpoints
|
| 16 |
|
| 17 |
-
This repository hosts the
|
| 18 |
|
| 19 |
- `multi-3dllm/`: MO3D, Shape Mating, and Change Captioning
|
| 20 |
- `multi-3dllm-classification/`: ModelNet40 zero-shot classification
|
| 21 |
|
| 22 |
-
|
|
|
|
|
|
|
| 23 |
|
| 24 |
```text
|
| 25 |
https://github.com/KohsukeIde/BeyondSingleObject
|
| 26 |
```
|
| 27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
## Download
|
| 29 |
|
| 30 |
```bash
|
|
@@ -37,18 +45,37 @@ Expected local layout:
|
|
| 37 |
|
| 38 |
```text
|
| 39 |
checkpoints/
|
| 40 |
-
|
| 41 |
-
|
| 42 |
data/
|
| 43 |
```
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
## Usage
|
| 46 |
|
| 47 |
-
Example inference
|
| 48 |
|
| 49 |
```bash
|
| 50 |
MODEL_PATH=checkpoints/multi-3dllm \
|
| 51 |
-
|
|
|
|
|
|
|
| 52 |
scripts/eval/infer.sh
|
| 53 |
```
|
| 54 |
|
|
@@ -64,15 +91,12 @@ TARGET_POSITION=1 \
|
|
| 64 |
scripts/eval/eval_modelnet.sh
|
| 65 |
```
|
| 66 |
|
| 67 |
-
Repeat `(NUM_OBJECTS, TARGET_POSITION) = (1,1), (2,1), (2,2), (3,1), (3,2),
|
| 68 |
-
(3,3)` for the full table.
|
| 69 |
-
|
| 70 |
## Notes
|
| 71 |
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
|
| 77 |
## License
|
| 78 |
|
|
|
|
| 14 |
|
| 15 |
# Multi-3DLLM Checkpoints
|
| 16 |
|
| 17 |
+
This repository hosts the BeyondSingleObject checkpoints:
|
| 18 |
|
| 19 |
- `multi-3dllm/`: MO3D, Shape Mating, and Change Captioning
|
| 20 |
- `multi-3dllm-classification/`: ModelNet40 zero-shot classification
|
| 21 |
|
| 22 |
+
The task checkpoint uses the Patch-Interaction Transformer implementation and
|
| 23 |
+
was jointly fine-tuned on the three released task datasets. Code and training
|
| 24 |
+
scripts are available at:
|
| 25 |
|
| 26 |
```text
|
| 27 |
https://github.com/KohsukeIde/BeyondSingleObject
|
| 28 |
```
|
| 29 |
|
| 30 |
+
Dataset annotations and point-cloud assets are available at:
|
| 31 |
+
|
| 32 |
+
```text
|
| 33 |
+
https://huggingface.co/datasets/idekoh/BeyondSingleObject
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
## Download
|
| 37 |
|
| 38 |
```bash
|
|
|
|
| 45 |
|
| 46 |
```text
|
| 47 |
checkpoints/
|
| 48 |
+
|-- multi-3dllm/
|
| 49 |
+
`-- multi-3dllm-classification/
|
| 50 |
data/
|
| 51 |
```
|
| 52 |
|
| 53 |
+
## Evaluation Snapshot
|
| 54 |
+
|
| 55 |
+
The following deterministic results use the compact evaluation subsets and the
|
| 56 |
+
public inference format. They are regression checks rather than replacements
|
| 57 |
+
for the paper's LLM-judged reasoning metrics.
|
| 58 |
+
|
| 59 |
+
| Task | Metric | Result |
|
| 60 |
+
| --- | --- | ---: |
|
| 61 |
+
| MO3D holistic QA | Binary accuracy | 84.0% (42/50) |
|
| 62 |
+
| Shape Mating | Selection accuracy | 73.0% (146/200) |
|
| 63 |
+
| Change Captioning verify | Binary accuracy | 67.0% (67/100) |
|
| 64 |
+
| Change Captioning verify, no point cloud | Binary accuracy | 50.0% (50/100) |
|
| 65 |
+
|
| 66 |
+
The separately released classification checkpoint follows the PointLLM
|
| 67 |
+
ModelNet40 evaluation protocol. See the repository README for its six
|
| 68 |
+
one-, two-, and three-object commands.
|
| 69 |
+
|
| 70 |
## Usage
|
| 71 |
|
| 72 |
+
Example inference:
|
| 73 |
|
| 74 |
```bash
|
| 75 |
MODEL_PATH=checkpoints/multi-3dllm \
|
| 76 |
+
ANNO_PATH=data/mo3d/test.json \
|
| 77 |
+
DATA_PATH=data/point_clouds \
|
| 78 |
+
OUTPUT_DIR=outputs/mo3d_eval \
|
| 79 |
scripts/eval/infer.sh
|
| 80 |
```
|
| 81 |
|
|
|
|
| 91 |
scripts/eval/eval_modelnet.sh
|
| 92 |
```
|
| 93 |
|
|
|
|
|
|
|
|
|
|
| 94 |
## Notes
|
| 95 |
|
| 96 |
+
LLM-judged reasoning and delta-caption scores depend on the judge model and
|
| 97 |
+
prompt configuration. Use the released evaluators and report the exact judge
|
| 98 |
+
configuration when comparing those metrics. The task and classification
|
| 99 |
+
checkpoints have different roles and should not be interchanged.
|
| 100 |
|
| 101 |
## License
|
| 102 |
|
multi-3dllm/config.json
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
"DEFAULT_POINT_END_TOKEN": "<point_end>",
|
| 3 |
"DEFAULT_POINT_PATCH_TOKEN": "<point_patch>",
|
| 4 |
"DEFAULT_POINT_START_TOKEN": "<point_start>",
|
| 5 |
-
"_name_or_path": "/groups/gag51404/ide/
|
| 6 |
"architectures": [
|
| 7 |
"PointLLMCVPRLlamaForCausalLM"
|
| 8 |
],
|
|
|
|
| 2 |
"DEFAULT_POINT_END_TOKEN": "<point_end>",
|
| 3 |
"DEFAULT_POINT_PATCH_TOKEN": "<point_patch>",
|
| 4 |
"DEFAULT_POINT_START_TOKEN": "<point_start>",
|
| 5 |
+
"_name_or_path": "/groups/gag51404/ide/PointLLM/outputs/PointLLM_train_stage1/PointLLM_train_stage1_original",
|
| 6 |
"architectures": [
|
| 7 |
"PointLLMCVPRLlamaForCausalLM"
|
| 8 |
],
|
multi-3dllm/point_proj.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 567885048
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6d26f93ca9978468f6a6a7f188497a09fe1cd2a86f60c527186d529fada4950f
|
| 3 |
size 567885048
|
multi-3dllm/pytorch_model-00001-of-00003.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 9878039688
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d64e5adb20cea7673e9f929aacdc37243a32cd72cefb08939e19f39efb9a6172
|
| 3 |
size 9878039688
|
multi-3dllm/pytorch_model-00002-of-00003.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 9894801940
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b4ee892feff102a765e9d723cf92beaeb20705f2ae1994dadae55838787afafb
|
| 3 |
size 9894801940
|
multi-3dllm/pytorch_model-00003-of-00003.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 8117611116
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31a6cc7fbbdbbbcd79a8170cc3100b86d5e1daa90acbfe749711b282dd9331e4
|
| 3 |
size 8117611116
|