microstructure-unet / checkpoint_manifest.json
WUT-AI-AI4Mat's picture
Upload model card and dataset checkpoints
26a9ecc verified
Raw History Blame Contribute Delete
2.77 kB
{
"model": "U-Net",
"source_code": "https://github.com/WUT-AI-AI4Mat/Segmentation-methods-evaluation-for-quantitative-microstructure-analysis",
"checkpoint_count": 7,
"checkpoints": [
{
"dataset": "Aachen-Heerlen",
"task": "binary_segmentation",
"num_classes": 2,
"fine_tuning_route": "full_model",
"path": "checkpoints/Aachen-Heerlen/Best_Model.pth",
"size_bytes": 130438093,
"sha256": "70f3d25e937de33b3aa7480df248f148f515179738051d623d9918ccf338c985",
"requires_base_checkpoint": "none"
},
{
"dataset": "EBC",
"task": "multiclass_semantic_segmentation",
"num_classes": 3,
"fine_tuning_route": "full_model",
"path": "checkpoints/EBC/Best_Model.pth",
"size_bytes": 130439245,
"sha256": "08fdd67970562adf9184486d56f7c1ed1abd6a17ac964b5297e29ee7dcdc3c3c",
"requires_base_checkpoint": "none"
},
{
"dataset": "EMPS",
"task": "binary_segmentation",
"num_classes": 2,
"fine_tuning_route": "full_model",
"path": "checkpoints/EMPS/Best_Model.pth",
"size_bytes": 130438093,
"sha256": "f53ce3d927ba10d5d898efd6aa9f78d7a85351057c824e3b4471addaa9e08008",
"requires_base_checkpoint": "none"
},
{
"dataset": "Grain",
"task": "binary_segmentation",
"num_classes": 2,
"fine_tuning_route": "full_model",
"path": "checkpoints/Grain/Best_Model.pth",
"size_bytes": 130438093,
"sha256": "1e8848fc872b36ef9d7c093477a5a92ada47dfb0e468da85e090ce1b36063fd1",
"requires_base_checkpoint": "none"
},
{
"dataset": "MetalDAM",
"task": "multiclass_semantic_segmentation",
"num_classes": 5,
"fine_tuning_route": "full_model",
"path": "checkpoints/MetalDAM/Best_Model.pth",
"size_bytes": 130440397,
"sha256": "f957f80232f02aa0a0b615db37412aadf0909ef959fb1defc8a0a351e33c3778",
"requires_base_checkpoint": "none"
},
{
"dataset": "Super",
"task": "multiclass_semantic_segmentation",
"num_classes": 3,
"fine_tuning_route": "full_model",
"path": "checkpoints/Super/Best_Model.pth",
"size_bytes": 130439245,
"sha256": "2e2800754c8045b361bf9b7c7c690aa83b5817315a225eaafe5f63e30a36490e",
"requires_base_checkpoint": "none"
},
{
"dataset": "UHCS",
"task": "multiclass_semantic_segmentation",
"num_classes": 7,
"fine_tuning_route": "full_model",
"path": "checkpoints/UHCS/Best_Model.pth",
"size_bytes": 130441549,
"sha256": "ce4392bba3bd6653311b2b925684bbb0758f7aa73336ebd78aa13b30370aa18d",
"requires_base_checkpoint": "none"
}
]
}