WUT-AI-AI4Mat commited on
Commit
145bfa6
·
verified ·
1 Parent(s): c1253e1

Upload model card and dataset checkpoints

Browse files
CHECKSUMS.sha256 ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ e9b58b033d2387cc2c2b2371044c9256dc60e676e68ece8660cbec8620d6b796 checkpoints/Aachen-Heerlen/best_mIoU_epoch_76.pth
2
+ 8e409a056d869907021f9bf208e16fe4a7a7b7c469c83ff79455a05552cb6392 checkpoints/EBC/best_mIoU_epoch_128.pth
3
+ d66d7ccf1e05ba21447a3a584fdcd1d8b8e0b87d7582d423e036077dd8f09d2d checkpoints/EMPS/best_mIoU_epoch_153.pth
4
+ 73b823393881630fc95b1a42cbfb25a8708da027bd694ddceda6d7fbba635b1c checkpoints/Grain/best_mIoU_epoch_63.pth
5
+ 619b95528d6510a50b8c67735900df371c532e4ff9871aeb25e85ccb1da32c45 checkpoints/MetalDAM/best_mIoU_epoch_96.pth
6
+ d7b101742362ad602998186ab6e28f7158b822862e85d9dcd5b6084774228167 checkpoints/Super/best_mIoU_epoch_170.pth
7
+ 59f2fcdbe48c0dedf4bacd80026a2e499f56a690a07e561d2acae320fb899736 checkpoints/UHCS/best_mIoU_epoch_137.pth
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2026 The Material Image Segmentation Benchmark Authors
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: mmsegmentation
4
+ pipeline_tag: image-segmentation
5
+ tags:
6
+ - semantic-segmentation
7
+ - microstructure
8
+ - materials-science
9
+ - pytorch
10
+ - segformer
11
+ - mmsegmentation
12
+ ---
13
+
14
+ # SegFormer-B0 for Quantitative Microstructure Segmentation
15
+
16
+ ## Model Description
17
+
18
+ This repository contains seven dataset-specific SegFormer checkpoints trained
19
+ with MMSegmentation. The architecture uses a MiT-B0 backbone and SegFormer
20
+ decode head with 512 x 512 inputs.
21
+
22
+ Source code and dataset-specific configs:
23
+ https://github.com/WUT-AI-AI4Mat/microstructure-segmentation-benchmark
24
+
25
+ ## Checkpoints
26
+
27
+ | Dataset | Classes | File |
28
+ | --- | ---: | --- |
29
+ | Aachen-Heerlen | 2 | `checkpoints/Aachen-Heerlen/best_mIoU_epoch_76.pth` |
30
+ | EMPS | 2 | `checkpoints/EMPS/best_mIoU_epoch_153.pth` |
31
+ | Grain | 2 | `checkpoints/Grain/best_mIoU_epoch_63.pth` |
32
+ | EBC | 3 | `checkpoints/EBC/best_mIoU_epoch_128.pth` |
33
+ | Super | 3 | `checkpoints/Super/best_mIoU_epoch_170.pth` |
34
+ | MetalDAM | 5 | `checkpoints/MetalDAM/best_mIoU_epoch_96.pth` |
35
+ | UHCS | 7 | `checkpoints/UHCS/best_mIoU_epoch_137.pth` |
36
+
37
+ Class counts include background.
38
+
39
+ ## Training Details
40
+
41
+ - Backbone: MiT-B0
42
+ - Input size: 512 x 512
43
+ - Training batch size: 32
44
+ - Validation and test batch size: 16
45
+ - Epochs: 200
46
+ - Optimizer: AdamW
47
+ - Learning rate: 0.00006
48
+ - Adam betas: 0.9 and 0.999
49
+ - Weight decay: 0.01
50
+ - Scheduler: five-epoch linear warm-up followed by polynomial decay
51
+ - Initialization: `mit_b0_20220624-7e0fe6dd.pth`
52
+
53
+ ## Usage
54
+
55
+ Install the MMSegmentation environment described in the source repository.
56
+ Example for EBC:
57
+
58
+ ```bash
59
+ hf download NAMESPACE/microstructure-segformer-b0 checkpoints/EBC/best_mIoU_epoch_128.pth --local-dir weights/segformer-b0
60
+ cd mmsegmentation
61
+ export DATASET_ROOT=/path/to/EBC
62
+ python tools/test_ebc_segformer.py \
63
+ --dataset-root /path/to/EBC \
64
+ --checkpoint ../weights/segformer-b0/checkpoints/EBC/best_mIoU_epoch_128.pth \
65
+ --result-root ../results/segformer_ebc
66
+ ```
67
+
68
+ Use the corresponding dataset config and `tools/test_*_segformer.py` wrapper
69
+ for the other checkpoints.
70
+
71
+ ## Evaluation
72
+
73
+ The shared testing code calls `Myutils.metrics.Metric.compute_all`, saves raw
74
+ prediction masks, and exports an Excel report. Reported metrics include mIoU,
75
+ Dice, precision, recall, accuracy, HD95, Hausdorff distance, NSD, MAE, MBSS,
76
+ and MBSS_add. Numerical benchmark results will be linked after the associated
77
+ paper becomes publicly available.
78
+
79
+ ## Intended Use and Limitations
80
+
81
+ The checkpoints are intended for reproducing segmentation experiments on the
82
+ named datasets. Performance may not transfer to unseen materials, imaging
83
+ modalities, magnifications, or annotation conventions. The models are not
84
+ validated for safety-critical or industrial quality-control decisions.
85
+
86
+ ## License and Citation
87
+
88
+ The released experiment files are provided under the MIT license.
89
+ MMSegmentation and the SegFormer initialization remain subject to their
90
+ upstream licenses. A paper citation will be added after publication.
checkpoint_manifest.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "SegFormer-B0",
3
+ "source_code": "https://github.com/WUT-AI-AI4Mat/microstructure-segmentation-benchmark",
4
+ "checkpoint_count": 7,
5
+ "checkpoints": [
6
+ {
7
+ "dataset": "Aachen-Heerlen",
8
+ "task": "binary_segmentation",
9
+ "num_classes": 2,
10
+ "fine_tuning_route": "full_model",
11
+ "path": "checkpoints/Aachen-Heerlen/best_mIoU_epoch_76.pth",
12
+ "size_bytes": 21396082,
13
+ "sha256": "e9b58b033d2387cc2c2b2371044c9256dc60e676e68ece8660cbec8620d6b796",
14
+ "requires_base_checkpoint": "none"
15
+ },
16
+ {
17
+ "dataset": "EBC",
18
+ "task": "multiclass_semantic_segmentation",
19
+ "num_classes": 3,
20
+ "fine_tuning_route": "full_model",
21
+ "path": "checkpoints/EBC/best_mIoU_epoch_128.pth",
22
+ "size_bytes": 15111922,
23
+ "sha256": "8e409a056d869907021f9bf208e16fe4a7a7b7c469c83ff79455a05552cb6392",
24
+ "requires_base_checkpoint": "none"
25
+ },
26
+ {
27
+ "dataset": "EMPS",
28
+ "task": "binary_segmentation",
29
+ "num_classes": 2,
30
+ "fine_tuning_route": "full_model",
31
+ "path": "checkpoints/EMPS/best_mIoU_epoch_153.pth",
32
+ "size_bytes": 18518578,
33
+ "sha256": "d66d7ccf1e05ba21447a3a584fdcd1d8b8e0b87d7582d423e036077dd8f09d2d",
34
+ "requires_base_checkpoint": "none"
35
+ },
36
+ {
37
+ "dataset": "Grain",
38
+ "task": "binary_segmentation",
39
+ "num_classes": 2,
40
+ "fine_tuning_route": "full_model",
41
+ "path": "checkpoints/Grain/best_mIoU_epoch_63.pth",
42
+ "size_bytes": 17518130,
43
+ "sha256": "73b823393881630fc95b1a42cbfb25a8708da027bd694ddceda6d7fbba635b1c",
44
+ "requires_base_checkpoint": "none"
45
+ },
46
+ {
47
+ "dataset": "MetalDAM",
48
+ "task": "multiclass_semantic_segmentation",
49
+ "num_classes": 5,
50
+ "fine_tuning_route": "full_model",
51
+ "path": "checkpoints/MetalDAM/best_mIoU_epoch_96.pth",
52
+ "size_bytes": 15183218,
53
+ "sha256": "619b95528d6510a50b8c67735900df371c532e4ff9871aeb25e85ccb1da32c45",
54
+ "requires_base_checkpoint": "none"
55
+ },
56
+ {
57
+ "dataset": "Super",
58
+ "task": "multiclass_semantic_segmentation",
59
+ "num_classes": 3,
60
+ "fine_tuning_route": "full_model",
61
+ "path": "checkpoints/Super/best_mIoU_epoch_170.pth",
62
+ "size_bytes": 15157746,
63
+ "sha256": "d7b101742362ad602998186ab6e28f7158b822862e85d9dcd5b6084774228167",
64
+ "requires_base_checkpoint": "none"
65
+ },
66
+ {
67
+ "dataset": "UHCS",
68
+ "task": "multiclass_semantic_segmentation",
69
+ "num_classes": 7,
70
+ "fine_tuning_route": "full_model",
71
+ "path": "checkpoints/UHCS/best_mIoU_epoch_137.pth",
72
+ "size_bytes": 18727026,
73
+ "sha256": "59f2fcdbe48c0dedf4bacd80026a2e499f56a690a07e561d2acae320fb899736",
74
+ "requires_base_checkpoint": "none"
75
+ }
76
+ ]
77
+ }
checkpoints/Aachen-Heerlen/best_mIoU_epoch_76.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9b58b033d2387cc2c2b2371044c9256dc60e676e68ece8660cbec8620d6b796
3
+ size 21396082
checkpoints/EBC/best_mIoU_epoch_128.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e409a056d869907021f9bf208e16fe4a7a7b7c469c83ff79455a05552cb6392
3
+ size 15111922
checkpoints/EMPS/best_mIoU_epoch_153.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d66d7ccf1e05ba21447a3a584fdcd1d8b8e0b87d7582d423e036077dd8f09d2d
3
+ size 18518578
checkpoints/Grain/best_mIoU_epoch_63.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73b823393881630fc95b1a42cbfb25a8708da027bd694ddceda6d7fbba635b1c
3
+ size 17518130
checkpoints/MetalDAM/best_mIoU_epoch_96.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:619b95528d6510a50b8c67735900df371c532e4ff9871aeb25e85ccb1da32c45
3
+ size 15183218
checkpoints/Super/best_mIoU_epoch_170.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7b101742362ad602998186ab6e28f7158b822862e85d9dcd5b6084774228167
3
+ size 15157746
checkpoints/UHCS/best_mIoU_epoch_137.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59f2fcdbe48c0dedf4bacd80026a2e499f56a690a07e561d2acae320fb899736
3
+ size 18727026