mattaq/GelGenie-AttUNet-CW-4-Nov-2023
By mattaq · ↓ 0 · ♥ 0 · Updated 2024-10-07
What's in this repo
This page lists the files in mattaq/GelGenie-AttUNet-CW-4-Nov-2023, including attunet_nov_4_class_weighting_epoch_338.onnx, attunet_nov_4_class_weighting_epoch_338_fixed_dim_1024.onnx.
- 2 ONNX models (.onnx files)
- 19 files, including 7 .pdf, 3 .txt, 2 .onnx, 1 .md, 1 .pth, 1 .toml
- Last updated 2024-10-07
ONNX files
| File | Size | Download |
|---|---|---|
| onnx_checkpoints/attunet_nov_4_class_weighting_epoch_338.onnx | — | Download |
| onnx_checkpoints/attunet_nov_4_class_weighting_epoch_338_fixed_dim_1024.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
This model is just a prototype used during the development process of GelGenie. It consists of a Attention UNet model (Monai) trained on the final GelGenie dataset.
Our codebase is fully open-sourced and is available here: https://github.com/mattaq31/GelGenie.
Model info
Created: 2023-11-05
Updated: 2024-10-07
View on Hugging Face