STELLAR-CLASS-5L-8-E2 (Residual CNN) - Fetal Planes Classification

This model is a custom 5-block convolutional network, trained from scratch (no ImageNet pretraining) on single-channel 128x128 fetal ultrasound images, to classify fetal ultrasound standard planes.

Model Performance (Test Set)

  • Accuracy: 0.2800
  • Macro F1 Score: 0.2585
  • Macro Precision: 0.3735
  • Macro Recall: 0.3382

##Changes made - lr =1e-5 - Added residual blocks in 3 -5 layers

Dataset Information

This model was trained on dataset Marc-HealthAI/combined_fetal_ultrasound_dataset.

  • Classes: Fetal abdomen, Fetal brain, Fetal femur, Fetal spine, Fetal thorax, Maternal Cervix, Other, Placenta
  • Dataset Features: ['image', 'label']

Dataset Description

No explicit description provided in dataset metadata.


Original Dataset README Content

Click to expand original dataset README
Downloads last month
5,810
Safetensors
Model size
4.91M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support