Clarify output labels and training mixture
Browse filesExplain aggregate spine annotations without presenting them as checkpoint channels, and direct readers to the paper for training mixture proportions.
README.md
CHANGED
|
@@ -141,11 +141,11 @@ resample back to the original image size.
|
|
| 141 |
|
| 142 |
## Output labels
|
| 143 |
|
| 144 |
-
|
| 145 |
-
|
| 146 |
-
|
| 147 |
-
|
| 148 |
-
|
| 149 |
|
| 150 |
- **Skull / shoulder girdle:** skull, scapulae, clavicles
|
| 151 |
- **Upper limb:** humeri, radii, ulnae, carpals, metacarpals, phalanges
|
|
@@ -168,17 +168,16 @@ architecture is in `config.yml`.
|
|
| 168 |
## Training data
|
| 169 |
|
| 170 |
The model was trained on three source types unified under the FleXray label
|
| 171 |
-
protocol
|
| 172 |
|
| 173 |
-
- **Real X-ray masks:** HandBones
|
| 174 |
-
|
| 175 |
-
- **Generated FluXray images
|
| 176 |
-
|
| 177 |
overlapping masks for every protocol structure.
|
| 178 |
-
- **Online CT->DRR rendering:** MOOSE / ENHANCE-PET 1.6k
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
with per-label attenuation jitter.
|
| 182 |
|
| 183 |
Training used AdamW (lr 3e-4, cosine schedule), a Dice + binary cross-entropy
|
| 184 |
loss routed per dataset (partially labeled sources ignore unlabeled channels),
|
|
|
|
| 141 |
|
| 142 |
## Output labels
|
| 143 |
|
| 144 |
+
FleXray outputs 60 foreground masks (61 channels including `background`). The
|
| 145 |
+
broader dataset/evaluation protocol also recognizes aggregate `lumbar_spine` and
|
| 146 |
+
`thoracolumbar_spine` annotations; these are evaluated by combining the relevant
|
| 147 |
+
per-vertebra outputs and are not checkpoint channels. Channel order is stored in
|
| 148 |
+
`label_schema.json`.
|
| 149 |
|
| 150 |
- **Skull / shoulder girdle:** skull, scapulae, clavicles
|
| 151 |
- **Upper limb:** humeri, radii, ulnae, carpals, metacarpals, phalanges
|
|
|
|
| 168 |
## Training data
|
| 169 |
|
| 170 |
The model was trained on three source types unified under the FleXray label
|
| 171 |
+
protocol. For training mixture proportions, please refer to the paper:
|
| 172 |
|
| 173 |
+
- **Real X-ray masks:** HandBones, FootBones, MURA forearm, and MURA humerus,
|
| 174 |
+
with our own annotations.
|
| 175 |
+
- **Generated FluXray images:** digitally reconstructed radiographs from the
|
| 176 |
+
MOOSE CTs, generatively edited toward real X-ray appearance, with exact
|
| 177 |
overlapping masks for every protocol structure.
|
| 178 |
+
- **Online CT->DRR rendering:** MOOSE / ENHANCE-PET 1.6k, Shoulder-CT, HaN-Seg,
|
| 179 |
+
PedsCT, RSNA cervical-spine fracture CTs, and ElbowCT, rendered to DRRs at
|
| 180 |
+
random poses during training with per-label attenuation jitter.
|
|
|
|
| 181 |
|
| 182 |
Training used AdamW (lr 3e-4, cosine schedule), a Dice + binary cross-entropy
|
| 183 |
loss routed per dataset (partially labeled sources ignore unlabeled channels),
|