ONNX 模型库
返回模型

说明文档

该模型由一个 SMP-UNet 组成,于 2023 年 12 月在 LSDB 数据集上为 GelGenie 训练。它仅应用于低分辨率图像。

该模型在训练集和验证集上共训练了 600 个 epoch,并提取了最终检查点(共 59 张图像)。

尽管训练使用了扩展的数据集,但其性能略逊于另一个可用的低分辨率模型。

如需了解训练所用配置的更多详情,请访问 https://huggingface.co/mattaq/GelGenie-LowRes-Extended-Dec-2023 并查看 config.toml 文件。我们的代码库已完全开源,可在此获取:https://github.com/mattaq31/GelGenie。

mattaq/GelGenie-LowRes-Extended-Dec-2023

作者 mattaq

image-segmentation pytorch
↓ 0 ♥ 0

创建时间: 2023-12-24 15:20:38+00:00

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

在 Hugging Face 上查看

文件 (12)

.gitattributes
README.md
checkpoints/checkpoint_epoch_600.pth
config.toml
model_structure.txt
model_summary.txt
onnx_checkpoints/unet_dec_21_lsdb_only_extended_set_epoch_600.onnx ONNX
onnx_checkpoints/unet_dec_21_lsdb_only_extended_set_epoch_600_fixed_dim_1024.onnx ONNX
time_log.txt
torchscript_checkpoints/unet_dec_21_lsdb_only_extended_set_epoch_600.pt
training_logs/metric_plots.pdf
training_logs/training_stats.csv