DeepLearnPhysics/PoLAr-MAE-Semantic

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.