Crimson Flame-6 (DenseNet161) - Fetal Planes Classification
This model is a fine-tuned DenseNet-161 trained to classify fetal ultrasound standard planes.
Model Performance (Test Set)
- Accuracy: 0.9515
- Macro F1 Score: 0.9403
- Macro Precision: 0.9334
- Macro Recall: 0.9498
Training History
| Epoch | Eval Loss | Eval Accuracy | Eval Precision (Macro) | Eval Recall (Macro) | Eval F1 (Macro) |
|---|---|---|---|---|---|
| 1.0 | 0.1656 | 0.9358 | 0.9130 | 0.9541 | 0.9307 |
| 2.0 | 0.1601 | 0.9370 | 0.9151 | 0.9547 | 0.9309 |
| 3.0 | 0.1483 | 0.9390 | 0.9210 | 0.9528 | 0.9340 |
| 4.0 | 0.1360 | 0.9305 | 0.9037 | 0.9565 | 0.9246 |
| 5.0 | 0.1366 | 0.9499 | 0.9300 | 0.9579 | 0.9425 |
| 6.0 | 0.1666 | 0.9556 | 0.9432 | 0.9562 | 0.9494 |
| 7.0 | 0.1628 | 0.9495 | 0.9357 | 0.9550 | 0.9447 |
| 8.0 | 0.2012 | 0.9568 | 0.9498 | 0.9518 | 0.9506 |
| 9.0 | 0.1747 | 0.9552 | 0.9423 | 0.9547 | 0.9481 |
| 10.0 | 0.1638 | 0.9572 | 0.9464 | 0.9604 | 0.9531 |
| 11.0 | 0.2389 | 0.9495 | 0.9533 | 0.9351 | 0.9438 |
| 12.0 | 0.1895 | 0.9535 | 0.9400 | 0.9526 | 0.9459 |
| 13.0 | 0.2364 | 0.9576 | 0.9526 | 0.9537 | 0.9529 |
| 14.0 | 0.2012 | 0.9455 | 0.9246 | 0.9594 | 0.9398 |
| 15.0 | 0.2759 | 0.9535 | 0.9521 | 0.9454 | 0.9485 |
| 16.0 | 0.2779 | 0.9560 | 0.9471 | 0.9522 | 0.9494 |
| 17.0 | 0.2405 | 0.9556 | 0.9428 | 0.9585 | 0.9501 |
| 18.0 | 0.2445 | 0.9523 | 0.9380 | 0.9576 | 0.9471 |
| 19.0 | 0.2345 | 0.9592 | 0.9520 | 0.9541 | 0.9530 |
| 20.0 | 0.2414 | 0.9531 | 0.9391 | 0.9539 | 0.9461 |
Dataset Information
This model was trained on dataset Marc-HealthAI/fetal-planes-classification-dataset-main.
- Classes: Fetal abdomen, Fetal brain, Fetal femur, Fetal thorax, Maternal cervix, Other
- Dataset Features:
['Image_name', 'Patient_num', 'Plane', 'Brain_plane', 'Operator', 'US_Machine', 'Split', 'image']
Dataset Description
No explicit description provided in dataset metadata.
Original Dataset README Content
Click to expand original dataset README
Fetal_Planes_DB
Burgos-Artizzu, X.P., Coronado-Gutiérrez, D., Valenzuela-Alcaraz, B. et al. Evaluation of deep convolutional neural networks for automatic classification of common maternal fetal ultrasound planes. Sci Rep 10, 10200 (2020). https://doi.org/10.1038/s41598-020-67076-5
Data Description
A large dataset of routinely acquired maternal-fetal screening ultrasound images collected from two different hospitals by several operators and ultrasound machines. All images were manually labeled by an expert maternal fetal clinician (B.V-A.). Images were divided into 6 classes: four of the most widely used fetal anatomical planes (Abdomen, Brain, Femur and Thorax), the mother’s cervix (widely used for prematurity screening) and a general category to include any other less common image plane. Fetal brain images were further categorized into the 3 most common fetal brain planes (Trans-thalamic, Trans-cerebellum, Trans-ventricular) to judge fine grain categorization performance. The final dataset is comprised of over 12,400 images from 1,792 patients.
The dataset details are described in our open-acces paper: Evaluation of deep convolutional neural networks for automatic classification of common maternal fetal ultrasound planes
If you find this dataset useful, please cite:
@article{Burgos-ArtizzuFetalPlanesDataset,
title={Evaluation of deep convolutional neural networks for automatic classification of common maternal fetal ultrasound planes},
author={Burgos-Artizzu, X.P. and Coronado-Gutiérrez, D. and Valenzuela-Alcaraz, B. and Bonet-Carne, E. and Eixarch, E. and Crispi, F. and Gratacós, E.},
journal={Nature Scientific Reports},
volume={10},
pages={10200},
doi="10.1038/s41598-020-67076-5",
year={2020}
}
- Downloads last month
- 11