PoLAr-MAE-Semantic / README.md
youngsm's picture
Update README.md
450d8ad verified
|
Raw
History Blame Contribute Delete
728 Bytes
metadata
library_name: pimm
datasets:
  - DeepLearnPhysics/PILArNet-M
tags:
  - particle-physics
  - lartpc
  - point-cloud

PoLAr-MAE — semantic segmentation

PoLAr-MAE fine-tuned for 4-class LArTPC semantic segmentation (shower, track, Michel, delta); reproduces the paper mF1 ≈ 0.82.

  • pimm type: PoLArMAE-SemSeg · 4 classes

Loading

import pimm
model = pimm.from_pretrained("hf://deeplearnphysics/polar-mae-semantic")

Provenance

Repackaged from the original PoLAr-MAE release checkpoints into the pimm export format. Inherits the source repo license.