WUT-AI-AI4Mat commited on
Commit
adb2378
·
verified ·
1 Parent(s): 948e5e4

Upload model card and dataset checkpoints

Browse files
CHECKSUMS.sha256 ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ b5b57ff689d343a1e50361e728aa5a6075d5b2c611732ed2638a7220c31dfb84 checkpoints/Aachen-Heerlen/DeepLabV3P_Best_Model.pth
2
+ a2f860d5bbfea1738da8f59f79ec59f7a9b1c77dd5887b7e441c0bd844197a9d checkpoints/EBC/DeepLabV3P_Best_Model.pth
3
+ 8cdffae1cb0a1618608c980531f5a58a4645c33ecc6f5e5e6abd3c0fdcf35352 checkpoints/EMPS/DeepLabV3P_Best_Model.pth
4
+ 6d4228297dc63271f256348d5920f8c5b686152ba0e41277ee90dfba6c091757 checkpoints/Grain/DeepLabV3P_Best_Model.pth
5
+ f44c9ec082114012fc287767f8c3b786d99f2deb46c4b62b55a46d01db806608 checkpoints/MetalDAM/DeepLabV3P_Best_Model.pth
6
+ 83b1c9404a285e6c657762ee7ad03e1458c5bf14f609cc1c81582825bbff9562 checkpoints/Super/DeepLabV3P_Best_Model.pth
7
+ 3a846d500e2c1ed875abd47f3ccc911dd26084bd6de66123bc4da8851b5d6823 checkpoints/UHCS/DeepLabV3P_Best_Model.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,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: segmentation_models_pytorch
4
+ pipeline_tag: image-segmentation
5
+ tags:
6
+ - semantic-segmentation
7
+ - microstructure
8
+ - materials-science
9
+ - pytorch
10
+ - deeplabv3plus
11
+ ---
12
+
13
+ # DeepLabV3+ for Quantitative Microstructure Segmentation
14
+
15
+ ## Model Description
16
+
17
+ This repository contains seven dataset-specific DeepLabV3+ checkpoints used
18
+ in a benchmark of microstructure segmentation methods. The implementation is
19
+ from `segmentation_models_pytorch` with a ResNet-50 encoder initialized from
20
+ ImageNet weights. Inputs are resized to 512 x 512 pixels and predictions are
21
+ restored to the original image size with nearest-neighbor interpolation.
22
+
23
+ Source code:
24
+ https://github.com/WUT-AI-AI4Mat/microstructure-segmentation-benchmark
25
+
26
+ ## Checkpoints
27
+
28
+ | Dataset | Task | Output channels | File |
29
+ | --- | --- | ---: | --- |
30
+ | Aachen-Heerlen | Binary segmentation | 1 | `checkpoints/Aachen-Heerlen/DeepLabV3P_Best_Model.pth` |
31
+ | EMPS | Binary segmentation | 1 | `checkpoints/EMPS/DeepLabV3P_Best_Model.pth` |
32
+ | Grain | Binary segmentation | 1 | `checkpoints/Grain/DeepLabV3P_Best_Model.pth` |
33
+ | EBC | 3-class segmentation | 3 | `checkpoints/EBC/DeepLabV3P_Best_Model.pth` |
34
+ | Super | 3-class segmentation | 3 | `checkpoints/Super/DeepLabV3P_Best_Model.pth` |
35
+ | MetalDAM | 5-class segmentation | 5 | `checkpoints/MetalDAM/DeepLabV3P_Best_Model.pth` |
36
+ | UHCS | 7-class segmentation | 7 | `checkpoints/UHCS/DeepLabV3P_Best_Model.pth` |
37
+
38
+ Binary checkpoints produce one foreground logit; background is represented by
39
+ the complementary binary label. Multiclass counts include background.
40
+
41
+ ## Training Details
42
+
43
+ - Input size: 512 x 512
44
+ - Batch size: 32
45
+ - Epochs: 500
46
+ - Optimizer: AdamW
47
+ - Learning rate: 0.0003
48
+ - Weight decay: 0.001
49
+ - Scheduler: CosineAnnealingLR with minimum learning rate 0.000001
50
+ - Early-stopping patience: 50
51
+ - Binary objective: Dice and BCEWithLogits
52
+ - Multiclass objective: Dice and cross-entropy
53
+
54
+ ## Usage
55
+
56
+ Example for binary EMPS segmentation:
57
+
58
+ ```bash
59
+ hf download NAMESPACE/microstructure-deeplabv3plus checkpoints/EMPS/DeepLabV3P_Best_Model.pth --local-dir weights/deeplabv3plus
60
+ python "CNN/DeepLabV3+/test.py" \
61
+ --dataset-root /path/to/EMPS \
62
+ --checkpoint weights/deeplabv3plus/checkpoints/EMPS/DeepLabV3P_Best_Model.pth \
63
+ --output-dir results/deeplabv3plus_emps
64
+ ```
65
+
66
+ For EBC, Super, MetalDAM, and UHCS, use
67
+ `CNN/DeepLabV3+/test_multi.py` and pass the matching `--num-classes` value.
68
+
69
+ ## Evaluation
70
+
71
+ The evaluation scripts save original-resolution masks and report mIoU, Dice,
72
+ precision, recall, accuracy, HD95, Hausdorff distance, NSD, MAE, MBSS, and
73
+ MBSS_add through `Myutils/metrics.py`. Numerical benchmark results will be
74
+ linked after the associated paper becomes publicly available.
75
+
76
+ ## Intended Use and Limitations
77
+
78
+ These checkpoints are intended for research reproduction and comparison on
79
+ the named microstructure datasets. They are not general-purpose segmentation
80
+ models. Domain shift in imaging, magnification, materials, or annotation
81
+ policy may reduce accuracy. They are not validated for safety-critical or
82
+ industrial quality-control decisions.
83
+
84
+ ## License and Citation
85
+
86
+ The released experiment files are provided under the MIT license. The model
87
+ implementation and ImageNet initialization remain subject to upstream terms.
88
+ A paper citation will be added after publication.
checkpoint_manifest.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "DeepLabV3+",
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/DeepLabV3P_Best_Model.pth",
12
+ "size_bytes": 107075735,
13
+ "sha256": "b5b57ff689d343a1e50361e728aa5a6075d5b2c611732ed2638a7220c31dfb84",
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/DeepLabV3P_Best_Model.pth",
22
+ "size_bytes": 107077783,
23
+ "sha256": "a2f860d5bbfea1738da8f59f79ec59f7a9b1c77dd5887b7e441c0bd844197a9d",
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/DeepLabV3P_Best_Model.pth",
32
+ "size_bytes": 107075735,
33
+ "sha256": "8cdffae1cb0a1618608c980531f5a58a4645c33ecc6f5e5e6abd3c0fdcf35352",
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/DeepLabV3P_Best_Model.pth",
42
+ "size_bytes": 107075735,
43
+ "sha256": "6d4228297dc63271f256348d5920f8c5b686152ba0e41277ee90dfba6c091757",
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/DeepLabV3P_Best_Model.pth",
52
+ "size_bytes": 107079831,
53
+ "sha256": "f44c9ec082114012fc287767f8c3b786d99f2deb46c4b62b55a46d01db806608",
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/DeepLabV3P_Best_Model.pth",
62
+ "size_bytes": 107077783,
63
+ "sha256": "83b1c9404a285e6c657762ee7ad03e1458c5bf14f609cc1c81582825bbff9562",
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/DeepLabV3P_Best_Model.pth",
72
+ "size_bytes": 107081879,
73
+ "sha256": "3a846d500e2c1ed875abd47f3ccc911dd26084bd6de66123bc4da8851b5d6823",
74
+ "requires_base_checkpoint": "none"
75
+ }
76
+ ]
77
+ }
checkpoints/Aachen-Heerlen/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5b57ff689d343a1e50361e728aa5a6075d5b2c611732ed2638a7220c31dfb84
3
+ size 107075735
checkpoints/EBC/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2f860d5bbfea1738da8f59f79ec59f7a9b1c77dd5887b7e441c0bd844197a9d
3
+ size 107077783
checkpoints/EMPS/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cdffae1cb0a1618608c980531f5a58a4645c33ecc6f5e5e6abd3c0fdcf35352
3
+ size 107075735
checkpoints/Grain/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d4228297dc63271f256348d5920f8c5b686152ba0e41277ee90dfba6c091757
3
+ size 107075735
checkpoints/MetalDAM/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f44c9ec082114012fc287767f8c3b786d99f2deb46c4b62b55a46d01db806608
3
+ size 107079831
checkpoints/Super/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83b1c9404a285e6c657762ee7ad03e1458c5bf14f609cc1c81582825bbff9562
3
+ size 107077783
checkpoints/UHCS/DeepLabV3P_Best_Model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a846d500e2c1ed875abd47f3ccc911dd26084bd6de66123bc4da8851b5d6823
3
+ size 107081879