zouinkhim's picture
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/
2c17b0b verified
|
Raw History Blame Contribute Delete
712 Bytes

setup_54_2d — 2D-slice mito student (experimental)

One z-slice in, one slice out: input float32 [1,1,1,N,N] → output float32 raw logits [1,3,1,N-88,N-88] (mito affinities z, y, x; threshold at 0). 16 nm, input norm uint8/255*2-1.

Distilled from the 3D model setup_54 (PyTorch: cellmap/mito-aff-resunet-setup-54): every 3D kernel collapsed to 2D (sum of depth taps) and fine-tuned to reproduce the 3D predictions slice by slice. Training step, per-file fidelity and sizes are in manifest.json (same schema as the 3D folders, per-axis input_size / output_size).

cellmap-flow serverless: Input size 1,N,N, Output size 1,N-88,N-88, voxel 16,16,16, Channels 3, Has channel axis true.