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
Safetensors
Model size
26.7M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train Marc-HealthAI/Crimson-Moon-6-DenseNet161