Add pimm export (repackaged from DeepLearnPhysics/panda)

README.md

@@ -1,3 +1,47 @@
 ---
-license: mit
+library_name: pimm
+tags:
+  - particle-physics
+  - lartpc
+  - point-cloud
+  - point-transformer-v3
 ---
+
+# Panda — particle panoptic detector
+
+A query-based panoptic detector that segments individual particle
+instances in LArTPC images with per-particle PID, over a PTv3 backbone with a
+masked cross-attention decoder.
+
+- **pimm type:** `detector-v4`
+- **queries:** 32 · **classes:** 6 (PID) · **stuff classes:** [5]
+- **decoder depth:** 3 · **hidden:** 256 · **heads:** 16 · MLP heads.
+
+## Loading
+
+```python
+import pimm
+model = pimm.from_pretrained("hf://deeplearnphysics/panda-particle")
+```
+
+The architecture and all hyper-parameters are carried in `training_config.json`,
+so no config file is needed. Weights are bitwise-identical to the original
+checkpoint.
+
+## Input
+
+A pimm `Point` dict for LArTPC (PILArNet) point clouds: `coord` (xyz) + `energy`,
+collected as `feat_keys=("coord", "energy")` (4 input channels). Coordinates are
+normalized in the model forward.
+
+## Notes
+
+- Config sets `enable_flash=True`. On CPU or non-FlashAttention setups, override
+  it (e.g. pass `enable_flash=False` to `from_pretrained`, or for detectors set it
+  on the `backbone`).
+
+## Provenance
+
+Repackaged from the original [`DeepLearnPhysics/panda`](https://github.com/DeepLearnPhysics/panda) checkpoints into the
+[`pimm`](https://github.com/youngsm/particle-imaging-models) export format. Inherits
+the license of the source repository.

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:305ef5440f9f9e7980f333823fe9f7bd12b7c941a031de1d16fe7f22ab9dee68
+size 380715132

training_config.json

@@ -0,0 +1,133 @@
+{
+  "model": {
+    "type": "detector-v4",
+    "labels": [
+      "particle"
+    ],
+    "num_queries": 32,
+    "num_classes": 6,
+    "use_stuff_head": true,
+    "stuff_classes": [
+      5
+    ],
+    "mlp_point_proj": true,
+    "full_in_channels": 1232,
+    "hidden_channels": 256,
+    "num_heads": 16,
+    "depth": 3,
+    "mlp_ratio": 4.0,
+    "qkv_bias": true,
+    "qk_scale": null,
+    "layer_scale": null,
+    "pre_norm": true,
+    "query_type": "learned",
+    "pos_emb": true,
+    "supervise_attn_mask": true,
+    "enable_flash": false,
+    "upcast_attention": false,
+    "upcast_softmax": false,
+    "backbone": {
+      "type": "PT-v3m2",
+      "in_channels": 4,
+      "order": [
+        "hilbert",
+        "hilbert-trans",
+        "z",
+        "z-trans"
+      ],
+      "stride": [
+        2,
+        2,
+        2,
+        2
+      ],
+      "enc_depths": [
+        3,
+        3,
+        3,
+        9,
+        3
+      ],
+      "enc_channels": [
+        48,
+        96,
+        192,
+        384,
+        512
+      ],
+      "enc_num_head": [
+        3,
+        6,
+        12,
+        24,
+        32
+      ],
+      "enc_patch_size": [
+        256,
+        256,
+        256,
+        256,
+        256
+      ],
+      "dec_depths": [
+        2,
+        2,
+        2,
+        2
+      ],
+      "dec_channels": [
+        64,
+        96,
+        192,
+        384
+      ],
+      "dec_num_head": [
+        4,
+        6,
+        12,
+        24
+      ],
+      "dec_patch_size": [
+        256,
+        256,
+        256,
+        256
+      ],
+      "mlp_ratio": 4,
+      "qkv_bias": true,
+      "qk_scale": null,
+      "layer_scale": 1e-05,
+      "attn_drop": 0.0,
+      "proj_drop": 0.0,
+      "drop_path": 0.3,
+      "shuffle_orders": true,
+      "pre_norm": true,
+      "enable_rpe": false,
+      "enable_flash": true,
+      "upcast_attention": false,
+      "upcast_softmax": false,
+      "traceable": false,
+      "mask_token": false,
+      "enc_mode": true,
+      "freeze_encoder": false
+    },
+    "criteria": [
+      {
+        "type": "FastUnifiedInstanceLoss",
+        "cost_mask": 1.0,
+        "cost_dice": 1.0,
+        "cost_class": 1.0,
+        "loss_weight_focal": 2.0,
+        "loss_weight_dice": 5.0,
+        "cls_weight_matched": 2.0,
+        "cls_weight_noobj": 0.5,
+        "focal_alpha": 0.25,
+        "focal_gamma": 2.0,
+        "aux_loss_weight": 1.0,
+        "num_points": 100000,
+        "truth_label": "instance",
+        "noobj_mask_loss_weight": 0.0
+      }
+    ]
+  }
+}