返回模型
说明文档
该模型由一个SMP-UNet组成，于2023年12月在LSDB数据集上为GelGenie训练完成。该模型仅适用于低分辨率图像。
该模型在训练集（53张图像）上训练了600个epoch，并根据验证集Dice分数提取了最佳检查点。
有关训练配置的更多详细信息，请访问 https://huggingface.co/mattaq/GelGenie-LowRes-Dec-2023 并查看 config.toml 文件。我们的代码库已完全开源，可在此处获取：https://github.com/mattaq31/GelGenie。
mattaq/GelGenie-LowRes-Dec-2023
作者 mattaq
image-segmentation
pytorch
↓ 0
♥ 0
创建时间: 2023-12-24 15:19:01+00:00
更新时间: 2024-10-07 03:38:59+00:00
在 Hugging Face 上查看文件 (18)
.gitattributes
README.md
checkpoints/checkpoint_epoch_427.pth
config.toml
model_structure.txt
model_summary.txt
onnx_checkpoints/unet_dec_21_lsdb_only_epoch_427.onnx
ONNX
onnx_checkpoints/unet_dec_21_lsdb_only_epoch_427_fixed_dim_1024.onnx
ONNX
segmentation_samples/sample_epoch_427_C107-3.pdf
segmentation_samples/sample_epoch_427_C53357.pdf
segmentation_samples/sample_epoch_427_C8-4.pdf
segmentation_samples/sample_epoch_600_C107-3.pdf
segmentation_samples/sample_epoch_600_C53357.pdf
segmentation_samples/sample_epoch_600_C8-4.pdf
time_log.txt
torchscript_checkpoints/unet_dec_21_lsdb_only_epoch_427.pt
training_logs/metric_plots.pdf
training_logs/training_stats.csv