返回模型
说明文档
该模型只是 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