Add FleXray base model bundle
Browse files- README.md +47 -0
- checksums.json +10 -0
- config.yml +87 -0
- label_schema.json +68 -0
- model.safetensors +3 -0
- preprocessing.json +12 -0
README.md
CHANGED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-nc-4.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- medical-image-segmentation
|
| 6 |
+
- x-ray
|
| 7 |
+
- flexray
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# VictorButoi/flexray-base
|
| 11 |
+
|
| 12 |
+
This is a FleXray segmentation model bundle. Fill in the public release details
|
| 13 |
+
before upload.
|
| 14 |
+
|
| 15 |
+
## Intended Use
|
| 16 |
+
|
| 17 |
+
Describe the intended research use, input image requirements, output labels, and
|
| 18 |
+
medical-use limitations. This model is not cleared for clinical diagnosis or
|
| 19 |
+
patient-care decisions.
|
| 20 |
+
|
| 21 |
+
## Training Data
|
| 22 |
+
|
| 23 |
+
Document every training dataset, source modality, license, preprocessing step,
|
| 24 |
+
and inclusion or exclusion criterion.
|
| 25 |
+
|
| 26 |
+
## Evaluation
|
| 27 |
+
|
| 28 |
+
Report evaluation datasets, metrics, confidence intervals where available, and
|
| 29 |
+
known failure modes.
|
| 30 |
+
|
| 31 |
+
## Files
|
| 32 |
+
|
| 33 |
+
- `model.safetensors`: exported model weights.
|
| 34 |
+
- `config.yml`: model architecture config.
|
| 35 |
+
- `label_schema.json`: ordered output labels.
|
| 36 |
+
- `preprocessing.json`: public preprocessing contract.
|
| 37 |
+
- `checksums.json`: SHA256 checksums.
|
| 38 |
+
|
| 39 |
+
## Licenses
|
| 40 |
+
|
| 41 |
+
Code license: MIT
|
| 42 |
+
|
| 43 |
+
Weights license: CC-BY-NC-4.0
|
| 44 |
+
|
| 45 |
+
## Citation
|
| 46 |
+
|
| 47 |
+
Cite FleXray and any datasets used to train this model.
|
checksums.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"algorithm": "sha256",
|
| 3 |
+
"files": {
|
| 4 |
+
"README.md": "59f705f7ab0923400d9b89bbfb91527e61e39883ab31655e44d6be017e3a00ef",
|
| 5 |
+
"config.yml": "ad2117f2f419aaeaa14d567cb92344293ab7533b2f5feaaf7a21b27844fdb955",
|
| 6 |
+
"label_schema.json": "01dede3e24ea1453852f3f7d908b05e15c75e6d9e942650888926e475d7176ad",
|
| 7 |
+
"model.safetensors": "93fe3640108dccd2b4fbe355fd55187613f0b609d1fb2c8d71957fb0f2f49ae7",
|
| 8 |
+
"preprocessing.json": "0ac2bb660c4d1a957e8537b0f14c7b3e451f89e21994a0f538d987135381d85a"
|
| 9 |
+
}
|
| 10 |
+
}
|
config.yml
ADDED
|
@@ -0,0 +1,87 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model:
|
| 2 |
+
_class: fxr.models.UNet
|
| 3 |
+
in_channels: 1
|
| 4 |
+
out_channels: 61
|
| 5 |
+
filters:
|
| 6 |
+
- 64
|
| 7 |
+
- 128
|
| 8 |
+
- 256
|
| 9 |
+
- 512
|
| 10 |
+
- 512
|
| 11 |
+
- 720
|
| 12 |
+
- 1024
|
| 13 |
+
convs_per_block: 2
|
| 14 |
+
bottleneck_convs: null
|
| 15 |
+
skip_connections: true
|
| 16 |
+
dropout: 0.0
|
| 17 |
+
norm: instance
|
| 18 |
+
residual: true
|
| 19 |
+
norm_after_activation: true
|
| 20 |
+
upsample_align_corners: true
|
| 21 |
+
residual_projection: auto
|
| 22 |
+
residual_shortcut_norm: true
|
| 23 |
+
protocol:
|
| 24 |
+
name: all_structures_flexray_v4
|
| 25 |
+
model_labels:
|
| 26 |
+
names:
|
| 27 |
+
- background
|
| 28 |
+
- skull
|
| 29 |
+
- scapulae
|
| 30 |
+
- clavicles
|
| 31 |
+
- humeri
|
| 32 |
+
- phalanges
|
| 33 |
+
- metacarpals
|
| 34 |
+
- carpals
|
| 35 |
+
- radii
|
| 36 |
+
- ulnae
|
| 37 |
+
- tibiae
|
| 38 |
+
- femurs
|
| 39 |
+
- patellae
|
| 40 |
+
- fibulae
|
| 41 |
+
- metatarsals
|
| 42 |
+
- tarsals
|
| 43 |
+
- toes
|
| 44 |
+
- rib_1
|
| 45 |
+
- rib_2
|
| 46 |
+
- rib_3
|
| 47 |
+
- rib_4
|
| 48 |
+
- rib_5
|
| 49 |
+
- rib_6
|
| 50 |
+
- rib_7
|
| 51 |
+
- rib_8
|
| 52 |
+
- rib_9
|
| 53 |
+
- rib_10
|
| 54 |
+
- rib_11
|
| 55 |
+
- rib_12
|
| 56 |
+
- sternum
|
| 57 |
+
- vertebra_c1
|
| 58 |
+
- vertebra_c2
|
| 59 |
+
- vertebra_c3
|
| 60 |
+
- vertebra_c4
|
| 61 |
+
- vertebra_c5
|
| 62 |
+
- vertebra_c6
|
| 63 |
+
- vertebra_c7
|
| 64 |
+
- vertebra_t1
|
| 65 |
+
- vertebra_t2
|
| 66 |
+
- vertebra_t3
|
| 67 |
+
- vertebra_t4
|
| 68 |
+
- vertebra_t5
|
| 69 |
+
- vertebra_t6
|
| 70 |
+
- vertebra_t7
|
| 71 |
+
- vertebra_t8
|
| 72 |
+
- vertebra_t9
|
| 73 |
+
- vertebra_t10
|
| 74 |
+
- vertebra_t11
|
| 75 |
+
- vertebra_t12
|
| 76 |
+
- vertebra_l1
|
| 77 |
+
- vertebra_l2
|
| 78 |
+
- vertebra_l3
|
| 79 |
+
- vertebra_l4
|
| 80 |
+
- vertebra_l5
|
| 81 |
+
- hips
|
| 82 |
+
- sacrum
|
| 83 |
+
- kidneys
|
| 84 |
+
- liver
|
| 85 |
+
- spleen
|
| 86 |
+
- lungs
|
| 87 |
+
- heart
|
label_schema.json
ADDED
|
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"label_names": [
|
| 3 |
+
"background",
|
| 4 |
+
"skull",
|
| 5 |
+
"scapulae",
|
| 6 |
+
"clavicles",
|
| 7 |
+
"humeri",
|
| 8 |
+
"phalanges",
|
| 9 |
+
"metacarpals",
|
| 10 |
+
"carpals",
|
| 11 |
+
"radii",
|
| 12 |
+
"ulnae",
|
| 13 |
+
"tibiae",
|
| 14 |
+
"femurs",
|
| 15 |
+
"patellae",
|
| 16 |
+
"fibulae",
|
| 17 |
+
"metatarsals",
|
| 18 |
+
"tarsals",
|
| 19 |
+
"toes",
|
| 20 |
+
"rib_1",
|
| 21 |
+
"rib_2",
|
| 22 |
+
"rib_3",
|
| 23 |
+
"rib_4",
|
| 24 |
+
"rib_5",
|
| 25 |
+
"rib_6",
|
| 26 |
+
"rib_7",
|
| 27 |
+
"rib_8",
|
| 28 |
+
"rib_9",
|
| 29 |
+
"rib_10",
|
| 30 |
+
"rib_11",
|
| 31 |
+
"rib_12",
|
| 32 |
+
"sternum",
|
| 33 |
+
"vertebra_c1",
|
| 34 |
+
"vertebra_c2",
|
| 35 |
+
"vertebra_c3",
|
| 36 |
+
"vertebra_c4",
|
| 37 |
+
"vertebra_c5",
|
| 38 |
+
"vertebra_c6",
|
| 39 |
+
"vertebra_c7",
|
| 40 |
+
"vertebra_t1",
|
| 41 |
+
"vertebra_t2",
|
| 42 |
+
"vertebra_t3",
|
| 43 |
+
"vertebra_t4",
|
| 44 |
+
"vertebra_t5",
|
| 45 |
+
"vertebra_t6",
|
| 46 |
+
"vertebra_t7",
|
| 47 |
+
"vertebra_t8",
|
| 48 |
+
"vertebra_t9",
|
| 49 |
+
"vertebra_t10",
|
| 50 |
+
"vertebra_t11",
|
| 51 |
+
"vertebra_t12",
|
| 52 |
+
"vertebra_l1",
|
| 53 |
+
"vertebra_l2",
|
| 54 |
+
"vertebra_l3",
|
| 55 |
+
"vertebra_l4",
|
| 56 |
+
"vertebra_l5",
|
| 57 |
+
"hips",
|
| 58 |
+
"sacrum",
|
| 59 |
+
"kidneys",
|
| 60 |
+
"liver",
|
| 61 |
+
"spleen",
|
| 62 |
+
"lungs",
|
| 63 |
+
"heart"
|
| 64 |
+
],
|
| 65 |
+
"model_id": "VictorButoi/flexray-base",
|
| 66 |
+
"num_labels": 61,
|
| 67 |
+
"weights_license": "CC-BY-NC-4.0"
|
| 68 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:93fe3640108dccd2b4fbe355fd55187613f0b609d1fb2c8d71957fb0f2f49ae7
|
| 3 |
+
size 286545060
|
preprocessing.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"color_mode": "grayscale",
|
| 3 |
+
"image_size": [
|
| 4 |
+
256,
|
| 5 |
+
256
|
| 6 |
+
],
|
| 7 |
+
"mean": null,
|
| 8 |
+
"pad_to_square": true,
|
| 9 |
+
"probability_mode": "multilabel",
|
| 10 |
+
"scale": "zero_one",
|
| 11 |
+
"std": null
|
| 12 |
+
}
|