ONNX 模型库
返回模型

说明文档

该模型只是 GelGenie 开发过程中使用的原型。它由一个在 GelGenie 最终数据集上训练的 Attention UNet 模型(Monai)组成。

我们的代码库已完全开源,可在此处获取:https://github.com/mattaq31/GelGenie。

mattaq/GelGenie-AttUNet-CW-4-Nov-2023

作者 mattaq

image-segmentation pytorch
↓ 0 ♥ 0

创建时间: 2023-11-05 11:08:07+00:00

更新时间: 2024-10-07 03:28:46+00:00

在 Hugging Face 上查看

文件 (19)

.gitattributes
README.md
checkpoints/checkpoint_epoch_338.pth
config.toml
model_structure.txt
model_summary.txt
onnx_checkpoints/attunet_nov_4_class_weighting_epoch_338.onnx ONNX
onnx_checkpoints/attunet_nov_4_class_weighting_epoch_338_fixed_dim_1024.onnx ONNX
onnx_checkpoints/visual_comparison_attunet_nov_4_class_weighting_epoch_338.png
segmentation_samples/sample_epoch_338_10.pdf
segmentation_samples/sample_epoch_338_18.pdf
segmentation_samples/sample_epoch_338_6.pdf
segmentation_samples/sample_epoch_465_10.pdf
segmentation_samples/sample_epoch_465_18.pdf
segmentation_samples/sample_epoch_465_6.pdf
time_log.txt
torchscript_checkpoints/attunet_nov_4_class_weighting_epoch_338.pt
training_logs/metric_plots.pdf
training_logs/training_stats.csv