ONNX 模型库
返回模型

说明文档

该模型只是 GelGenie 开发过程中使用的原型。它由在最终 GelGenie 数据集上训练的 SMP UNet 模型组成。更多详情请参见配置文件。

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

mattaq/GelGenie-SMPUNet-GP-4-Nov-2023

作者 mattaq

image-segmentation pytorch
↓ 0 ♥ 0

创建时间: 2023-11-05 11:14:55+00:00

更新时间: 2024-10-07 03:34:55+00:00

在 Hugging Face 上查看

文件 (22)

.gitattributes
README.md
checkpoints/checkpoint_epoch_122.pth
config.toml
model_structure.txt
model_summary.txt
onnx_checkpoints/unet_global_padding_nov_4_epoch_122.onnx ONNX
onnx_checkpoints/unet_global_padding_nov_4_epoch_198.onnx ONNX
onnx_checkpoints/unet_global_padding_nov_4_epoch_198_fixed_dim_1024.onnx ONNX
onnx_checkpoints/visual_comparison_unet_global_padding_nov_4_epoch_122.png
onnx_checkpoints/visual_comparison_unet_global_padding_nov_4_epoch_198.png
segmentation_samples/sample_epoch_122_C107-3.pdf
segmentation_samples/sample_epoch_122_C53357.pdf
segmentation_samples/sample_epoch_122_C8-4.pdf
segmentation_samples/sample_epoch_148_C107-3.pdf
segmentation_samples/sample_epoch_148_C53357.pdf
segmentation_samples/sample_epoch_148_C8-4.pdf
time_log.txt
torchscript_checkpoints/unet_global_padding_nov_4_epoch_122.pt
torchscript_checkpoints/unet_global_padding_nov_4_epoch_198.pt
training_logs/metric_plots.pdf
training_logs/training_stats.csv