ONNX 模型库
返回模型

说明文档

该模型是一个基于2023年12月通用模型进行微调的SMP-UNet。微调使用了包含清晰凝胶条带的20张凝胶图像(Stella数据集),训练了11个epoch。该模型牺牲了少量的通用性能,以在清晰、非常锐利的凝胶条带上获得更优异的结果。

如需了解训练配置的更多详细信息,请访问 https://huggingface.co/mattaq/GelGenie-Universal-FineTune-May-2024 并查看 config.toml 文件。我们的代码库完全开源,可在此获取:https://github.com/mattaq31/GelGenie。

该模型的 BioImage.IO 版本也可在此下载。

mattaq/GelGenie-Universal-FineTune-May-2024

作者 mattaq

image-segmentation pytorch
↓ 0 ♥ 0

创建时间: 2024-05-18 23:36:45+00:00

更新时间: 2025-01-14 04:18:06+00:00

在 Hugging Face 上查看

文件 (11)

.gitattributes
README.md
checkpoints/checkpoint_epoch_590.pth
checkpoints/checkpoint_epoch_600.pth
config.toml
gelgenie_finetuned_model_bioimageio.zip
onnx_checkpoints/unet_dec_21_finetune_epoch_590.onnx ONNX
onnx_checkpoints/unet_dec_21_finetune_epoch_590_fixed_dim_1024.onnx ONNX
torchscript_checkpoints/unet_dec_21_finetune_epoch_590.pt
training_logs/metric_plots.pdf
training_logs/training_stats.csv