README.md

metadata

language: en
license: lgpl-3.0
tags:
  - pytorch
  - torchscript
  - image-segmentation
  - particle-physics
  - neutrino-detectors
  - liquid-argon
datasets:
  - custom
library_name: pytorch
pipeline_tag: image-segmentation
model-index:
  - name: DNN ROI Models
    results:
      - task:
          type: image-segmentation
          name: Region of Interest Detection
        metrics:
          - type: iou
            value: N/A
            name: Intersection over Union
          - type: dice
            value: N/A
            name: Dice Coefficient
models:
  - icarus/moon-2025-08-25
  - pdhd/dikshant/mobileunet
  - pdhd/hokyeong/mobilenetv3
  - pdsp/unet
  - pdsp/nestedunet
  - sbnd/sbnd_data-v01_34_00

DNN ROI Models

A collection of deep neural network models for region of interest (ROI) detection in LArTPC experiments, including ICARUS, ProtoDUNE-HD, ProtoDUNE-SP, and SBND.

├── icarus
│   └── moon-2025-08-25
│       ├── plane0_rand.ts
│       └── plane1_rand.ts
├── pdhd
│   ├── dikshant
│   │   ├── mobileunet_largedataset_fullimage.ts
│   │   ├── mobileunet_largedataset_rebin4.ts
│   │   ├── unet_largedataset_fullimage.ts
│   │   └── unet_largedataset_rebin4.ts
│   └── hokyeong
│       └── CP49_mobilenetv3.ts
├── pdsp
│   ├── pth-model # models in pytorch pickle format
│   │   ├── nestedunet-l23-cosmic500-e50.pth # input: loose, MP2, MP3
│   │   ├── unet-l23-cosmic500-e50.pth # input: loose, MP2, MP3
│   │   └── unet-lt-cosmic500-e50.pth # input: loose, tight
│   ├── ts-model-1.3 # TorchScript model saved using PyTorch 1.3
│   │   ├── nestedunet-l23-cosmic500-e50.ts
│   │   ├── unet-l23-cosmic500-e50.ts
│   │   └── unet-lt-cosmic500-e50.ts
│   └── ts-model-2.3 # TorchScript model saved using PyTorch 2.3
│       ├── nestedunet-l23-cosmic500-e50.ts
│       ├── unet-l23-cosmic500-e50.ts
│       └── unet-lt-cosmic500-e50.ts
├── README.md
└── sbnd
    └── sbnd_data-v01_34_00
        ├── plane0.ts
        └── plane1.ts