Hugging Face's logo

cellmap
/
onnx-web-models

ONNX
electron-microscopy
segmentation
cellmap
webgpu
onnxruntime-web
Model card Files
xet
 Community
onnx-web-models / setup_55

Commit History

5 days ago

  1. 3D full finetuning: NDHWC train_graph.json + torch_params.json, structured weight_map to_onnx, 3D runtime; 2D spike peak_mb corrected

    zouinkhim
    Verified signature
    3331842

6 days ago

  1. Browser finetuning: manifest finetune block (head; full/tfjs for 2D students), initializer offsets, weight maps, torch_version/exporter/opset, source_repo for 2D students

    zouinkhim
    Verified signature
    1a480eb

7 days ago

  1. 2D students to top-level setup_53_2d/ setup_54_2d/ (3D manifest schema, per-axis sizes, step 20000); source_repo on 3D manifests; remove 2d/

    zouinkhim
    Verified signature
    2c17b0b

9 days ago

  1. All models: export Conv3d as exact Conv2d over depth-folded batch (~16x faster on onnxruntime-web WebGPU); drop experimental/

    zouinkhim
    Verified signature
    74ccc9b

  2. Unet3dIso: export 3D ConvTranspose as exact 1x1x1 Conv + depth-to-space (WebGPU EP has no 3D ConvTranspose)

    zouinkhim
    Verified signature
    6a5289f

  3. Add exp_mito ONNX web models (6 setups x 3 block sizes x fp32/fp16/int8w/int8w_fp16)

    zouinkhim
    Verified signature
    5c02968