SAVRN
Search Contact SAVRN

Open-weight model · Image classification

Crimson-Flame-6-resNet18

by AI/ML Mentorship Marc-HealthAI/Crimson-Flame-6-resNet18

Crimson-Flame-6-resNet18 is an open-weight model for image classification from AI/ML Mentorship, released under MIT License. It has 11M parameters. At 16-bit it needs about 0 GB of GPU memory, which fits on 1x MI300X from $1.85 an hour, at the lowest prices in the SAVRN Index. It draws 17 downloads a month.

This model is a fine-tuned ResNet-18 trained to classify fetal ultrasound standard planes. This model was trained on dataset Marc-HealthAI/fetal-planes-classification-dataset-main. No explicit description provided in dataset metadata.

Parameters11M
Context—
Weights44.8 MB
Licensemit
AccessOpen weights
Monthly Downloads17

Runs On

What it takes to serve Crimson-Flame-6-resNet18 (11M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Oct 5, 2026.

Crimson-Flame-6-resNet18 on every accelerator the SAVRN Index prices, at every precision

Model Card

By AI/ML Mentorship, published under mit, revision 411aadd37f4d.

This model is a fine-tuned ResNet-18 trained to classify fetal ultrasound standard planes. This model was trained on dataset Marc-HealthAI/fetal-planes-classification-dataset-main. No explicit description provided in dataset metadata. 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 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…

Read AI/ML Mentorship's full model card

Crimson Flame-6 (ResNet18) - Fetal Planes Classification

This model is a fine-tuned ResNet-18 trained to classify fetal ultrasound standard planes.

Model Performance (Test Set)

  • Accuracy: 0.9438
  • Macro F1 Score: 0.9332
  • Macro Precision: 0.9259
  • Macro Recall: 0.9430

Training History

Epoch Eval Loss Eval Accuracy Eval Precision (Macro) Eval Recall (Macro) Eval F1 (Macro)
1.0 0.2145 0.9184 0.8942 0.9394 0.9130
2.0 0.1825 0.9341 0.9122 0.9455 0.9271
3.0 0.1793 0.9301 0.9137 0.9438 0.9258
4.0 0.1710 0.9325 0.9174 0.9458 0.9290
5.0 0.1618 0.9337 0.9190 0.9446 0.9307
6.0 0.1672 0.9418 0.9280 0.9493 0.9377
7.0 0.1711 0.9414 0.9238 0.9566 0.9381
8.0 0.2063 0.9503 0.9466 0.9432 0.9431
9.0 0.1635 0.9406 0.9223 0.9525 0.9359
10.0 0.2121 0.9366 0.9194 0.9471 0.9319
11.0 0.1645 0.9572 0.9491 0.9572 0.9530
12.0 0.1618 0.9531 0.9375 0.9579 0.9471
13.0 0.1864 0.9499 0.9339 0.9527 0.9426
14.0 0.1858 0.9515 0.9360 0.9555 0.9451
15.0 0.2731 0.9531 0.9530 0.9484 0.9501
16.0 0.2352 0.9523 0.9420 0.9520 0.9467
17.0 0.1957 0.9463 0.9297 0.9545 0.9411

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](https://rdcu.be/b47NX) 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} }

Identity and Version

Repository
Marc-HealthAI/Crimson-Flame-6-resNet18
Publisher
AI/ML Mentorship
Task
Image classification
Modality
Image
Library
pytorch
Parameters
11M parameters
Languages
en
Revision
411aadd37f4d517136741eaadd254e4700029179
First published
2026-09-21
Last updated
2026-10-03

Files and Weights

153 files, 55.2 MB in total. The weights are 1 file totalling 44.8 MB in safetensors.

Weights1 file · 44.8 MB
Configuration1 file · 525 B
Documentation1 file · 4.1 KB
Other149 files · 10.5 MB
Repository1 file · 2.6 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights44.8 MB c3aefc5515df
config.jsonConfiguration525 B —
README.mdDocumentation4.1 KB —
CrimsonFlameconfusion_matrix.csvOther236 B —
CrimsonFlameconfusion_matrix.pngOther186.1 KB 889f306451f8
Crimson_Flameclassification_report.txtOther572 B —
accuracy_curve.pngOther101.6 KB 0a5d4416a62e
gradcam_eval_heatmaps/gradcam_grid_display_batch_1.pngOther989.1 KB b670415dccee
gradcam_eval_heatmaps/gradcam_grid_display_batch_2.pngOther1.0 MB 8f86d5b90627
gradcam_eval_heatmaps/gradcam_grid_display_batch_3.pngOther1.0 MB 5be3fa40d367
gradcam_eval_heatmaps/gradcam_grid_display_batch_4.pngOther995.5 KB d01a945ae2dc
gradcam_eval_heatmaps/gradcam_grid_display_batch_5.pngOther1.0 MB b162820f5071
gradcam_eval_heatmaps/per_class/gradcam_class_0_Fetal_abdomen.pngOther533.2 KB fa45de051a6e
gradcam_eval_heatmaps/per_class/gradcam_class_1_Fetal_brain.pngOther416.8 KB a5a2b2c4476f
gradcam_eval_heatmaps/per_class/gradcam_class_2_Fetal_femur.pngOther531.8 KB 91eebdb65d0a
gradcam_eval_heatmaps/per_class/gradcam_class_3_Fetal_thorax.pngOther519.2 KB b045f29b6035
gradcam_eval_heatmaps/sample_101_true_Fetal_femur_pred_Fetal_femur.pngOther21.7 KB —
gradcam_eval_heatmaps/sample_107_true_Fetal_brain_pred_Fetal_brain.pngOther22.4 KB —
gradcam_eval_heatmaps/sample_107_true_Fetal_brain_pred_Fetal_femur.pngOther23.0 KB —
gradcam_eval_heatmaps/sample_110_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther24.6 KB —
gradcam_eval_heatmaps/sample_111_true_Fetal_brain_pred_Fetal_brain.pngOther23.1 KB —
gradcam_eval_heatmaps/sample_112_true_Fetal_brain_pred_Fetal_brain.pngOther23.7 KB —
gradcam_eval_heatmaps/sample_114_true_Fetal_femur_pred_Fetal_femur.pngOther21.5 KB —
gradcam_eval_heatmaps/sample_116_true_Fetal_thorax_pred_Other.pngOther22.2 KB —
gradcam_eval_heatmaps/sample_119_true_Fetal_femur_pred_Fetal_femur.pngOther23.1 KB —
gradcam_eval_heatmaps/sample_125_true_Fetal_thorax_pred_Other.pngOther24.3 KB —
gradcam_eval_heatmaps/sample_12_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther25.2 KB —
gradcam_eval_heatmaps/sample_12_true_Fetal_abdomen_pred_Other.pngOther25.3 KB —
gradcam_eval_heatmaps/sample_132_true_Fetal_femur_pred_Fetal_femur.pngOther21.6 KB —
gradcam_eval_heatmaps/sample_135_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther24.4 KB —
gradcam_eval_heatmaps/sample_135_true_Fetal_abdomen_pred_Other.pngOther24.3 KB —
gradcam_eval_heatmaps/sample_136_true_Fetal_brain_pred_Fetal_brain.pngOther25.2 KB —
gradcam_eval_heatmaps/sample_13_true_Fetal_brain_pred_Fetal_brain.pngOther25.2 KB —
gradcam_eval_heatmaps/sample_140_true_Fetal_thorax_pred_Fetal_thorax.pngOther21.9 KB —
gradcam_eval_heatmaps/sample_142_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther23.3 KB —
gradcam_eval_heatmaps/sample_144_true_Fetal_thorax_pred_Other.pngOther22.9 KB —
gradcam_eval_heatmaps/sample_148_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther25.1 KB —
gradcam_eval_heatmaps/sample_150_true_Fetal_thorax_pred_Fetal_thorax.pngOther25.3 KB —
gradcam_eval_heatmaps/sample_150_true_Fetal_thorax_pred_Other.pngOther25.6 KB —
gradcam_eval_heatmaps/sample_155_true_Fetal_femur_pred_Fetal_femur.pngOther23.0 KB —
gradcam_eval_heatmaps/sample_15_true_Fetal_thorax_pred_Fetal_brain.pngOther23.3 KB —
gradcam_eval_heatmaps/sample_15_true_Fetal_thorax_pred_Fetal_thorax.pngOther23.6 KB —
gradcam_eval_heatmaps/sample_16_true_Fetal_brain_pred_Fetal_brain.pngOther18.8 KB —
gradcam_eval_heatmaps/sample_172_true_Fetal_brain_pred_Fetal_brain.pngOther20.9 KB —
gradcam_eval_heatmaps/sample_174_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther25.5 KB —
gradcam_eval_heatmaps/sample_174_true_Fetal_abdomen_pred_Other.pngOther25.9 KB —
gradcam_eval_heatmaps/sample_176_true_Fetal_brain_pred_Fetal_brain.pngOther24.5 KB —
gradcam_eval_heatmaps/sample_181_true_Fetal_abdomen_pred_Other.pngOther23.9 KB —
gradcam_eval_heatmaps/sample_183_true_Fetal_thorax_pred_Other.pngOther23.4 KB —
gradcam_eval_heatmaps/sample_185_true_Fetal_thorax_pred_Other.pngOther22.8 KB —
gradcam_eval_heatmaps/sample_186_true_Fetal_femur_pred_Fetal_femur.pngOther21.9 KB —
gradcam_eval_heatmaps/sample_186_true_Fetal_femur_pred_Other.pngOther22.0 KB —
gradcam_eval_heatmaps/sample_189_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther23.7 KB —
gradcam_eval_heatmaps/sample_189_true_Fetal_abdomen_pred_Other.pngOther23.9 KB —
gradcam_eval_heatmaps/sample_193_true_Fetal_femur_pred_Fetal_femur.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_193_true_Fetal_femur_pred_Other.pngOther20.6 KB —
gradcam_eval_heatmaps/sample_194_true_Fetal_thorax_pred_Fetal_thorax.pngOther24.5 KB —
gradcam_eval_heatmaps/sample_194_true_Fetal_thorax_pred_Other.pngOther23.8 KB —
gradcam_eval_heatmaps/sample_197_true_Fetal_brain_pred_Other.pngOther25.5 KB —
gradcam_eval_heatmaps/sample_207_true_Fetal_thorax_pred_Fetal_thorax.pngOther20.9 KB —
gradcam_eval_heatmaps/sample_20_true_Fetal_abdomen_pred_Other.pngOther19.2 KB —
gradcam_eval_heatmaps/sample_214_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther22.1 KB —
gradcam_eval_heatmaps/sample_215_true_Fetal_brain_pred_Fetal_brain.pngOther20.7 KB —
gradcam_eval_heatmaps/sample_216_true_Fetal_thorax_pred_Other.pngOther20.2 KB —
gradcam_eval_heatmaps/sample_229_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther21.1 KB —
gradcam_eval_heatmaps/sample_22_true_Fetal_thorax_pred_Other.pngOther17.3 KB —
gradcam_eval_heatmaps/sample_232_true_Fetal_abdomen_pred_Other.pngOther21.4 KB —
gradcam_eval_heatmaps/sample_235_true_Fetal_brain_pred_Fetal_brain.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_236_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_236_true_Fetal_abdomen_pred_Other.pngOther21.4 KB —
gradcam_eval_heatmaps/sample_23_true_Fetal_femur_pred_Fetal_femur.pngOther17.6 KB —
gradcam_eval_heatmaps/sample_244_true_Fetal_thorax_pred_Other.pngOther21.0 KB —
gradcam_eval_heatmaps/sample_248_true_Fetal_abdomen_pred_Other.pngOther20.8 KB —
gradcam_eval_heatmaps/sample_258_true_Fetal_brain_pred_Fetal_brain.pngOther21.8 KB —
gradcam_eval_heatmaps/sample_261_true_Fetal_thorax_pred_Fetal_thorax.pngOther22.9 KB —
gradcam_eval_heatmaps/sample_273_true_Fetal_femur_pred_Fetal_femur.pngOther21.1 KB —
gradcam_eval_heatmaps/sample_274_true_Fetal_thorax_pred_Other.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_279_true_Fetal_thorax_pred_Fetal_thorax.pngOther18.6 KB —
gradcam_eval_heatmaps/sample_282_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_287_true_Fetal_femur_pred_Fetal_femur.pngOther19.4 KB —
gradcam_eval_heatmaps/sample_287_true_Fetal_femur_pred_Other.pngOther19.4 KB —
gradcam_eval_heatmaps/sample_295_true_Fetal_thorax_pred_Other.pngOther21.0 KB —
gradcam_eval_heatmaps/sample_298_true_Fetal_brain_pred_Fetal_brain.pngOther19.7 KB —
gradcam_eval_heatmaps/sample_301_true_Fetal_femur_pred_Fetal_femur.pngOther19.5 KB —
gradcam_eval_heatmaps/sample_302_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther21.0 KB —
gradcam_eval_heatmaps/sample_308_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther23.2 KB —
gradcam_eval_heatmaps/sample_309_true_Fetal_femur_pred_Fetal_femur.pngOther19.9 KB —
gradcam_eval_heatmaps/sample_309_true_Fetal_femur_pred_Other.pngOther20.4 KB —
gradcam_eval_heatmaps/sample_311_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther22.3 KB —
gradcam_eval_heatmaps/sample_316_true_Fetal_brain_pred_Fetal_brain.pngOther21.7 KB —
gradcam_eval_heatmaps/sample_321_true_Fetal_brain_pred_Fetal_brain.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_325_true_Fetal_brain_pred_Fetal_brain.pngOther22.0 KB —
gradcam_eval_heatmaps/sample_327_true_Fetal_brain_pred_Fetal_brain.pngOther20.6 KB —
gradcam_eval_heatmaps/sample_331_true_Fetal_femur_pred_Fetal_femur.pngOther21.2 KB —
gradcam_eval_heatmaps/sample_331_true_Fetal_femur_pred_Other.pngOther20.8 KB —
gradcam_eval_heatmaps/sample_332_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther24.4 KB —
gradcam_eval_heatmaps/sample_332_true_Fetal_abdomen_pred_Other.pngOther24.8 KB —
gradcam_eval_heatmaps/sample_338_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther20.4 KB —
gradcam_eval_heatmaps/sample_343_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther21.9 KB —
gradcam_eval_heatmaps/sample_346_true_Fetal_brain_pred_Fetal_brain.pngOther21.1 KB —
gradcam_eval_heatmaps/sample_349_true_Fetal_femur_pred_Fetal_femur.pngOther20.0 KB —
gradcam_eval_heatmaps/sample_357_true_Fetal_brain_pred_Fetal_brain.pngOther20.9 KB —
gradcam_eval_heatmaps/sample_359_true_Fetal_femur_pred_Fetal_femur.pngOther21.1 KB —
gradcam_eval_heatmaps/sample_35_true_Fetal_brain_pred_Fetal_brain.pngOther20.1 KB —
gradcam_eval_heatmaps/sample_360_true_Fetal_brain_pred_Fetal_brain.pngOther20.8 KB —
gradcam_eval_heatmaps/sample_366_true_Fetal_abdomen_pred_Other.pngOther20.3 KB —
gradcam_eval_heatmaps/sample_368_true_Fetal_femur_pred_Fetal_femur.pngOther19.0 KB —
gradcam_eval_heatmaps/sample_368_true_Fetal_femur_pred_Other.pngOther19.1 KB —
gradcam_eval_heatmaps/sample_36_true_Fetal_femur_pred_Fetal_femur.pngOther20.5 KB —
gradcam_eval_heatmaps/sample_373_true_Fetal_brain_pred_Fetal_brain.pngOther21.7 KB —
gradcam_eval_heatmaps/sample_377_true_Fetal_brain_pred_Fetal_brain.pngOther20.2 KB —
gradcam_eval_heatmaps/sample_379_true_Fetal_brain_pred_Fetal_brain.pngOther21.0 KB —
gradcam_eval_heatmaps/sample_386_true_Fetal_brain_pred_Fetal_brain.pngOther15.5 KB —
gradcam_eval_heatmaps/sample_388_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther17.8 KB —
gradcam_eval_heatmaps/sample_388_true_Fetal_abdomen_pred_Other.pngOther18.1 KB —
gradcam_eval_heatmaps/sample_390_true_Fetal_brain_pred_Fetal_brain.pngOther19.8 KB —
gradcam_eval_heatmaps/sample_3_true_Fetal_thorax_pred_Fetal_brain.pngOther26.7 KB —
gradcam_eval_heatmaps/sample_400_true_Fetal_femur_pred_Fetal_femur.pngOther19.9 KB —
gradcam_eval_heatmaps/sample_401_true_Fetal_brain_pred_Fetal_brain.pngOther20.8 KB —
gradcam_eval_heatmaps/sample_407_true_Fetal_brain_pred_Fetal_brain.pngOther20.4 KB —
gradcam_eval_heatmaps/sample_40_true_Fetal_brain_pred_Fetal_brain.pngOther16.3 KB —
gradcam_eval_heatmaps/sample_412_true_Fetal_femur_pred_Fetal_femur.pngOther20.5 KB —
gradcam_eval_heatmaps/sample_414_true_Fetal_brain_pred_Fetal_brain.pngOther18.6 KB —
gradcam_eval_heatmaps/sample_415_true_Fetal_femur_pred_Fetal_brain.pngOther21.6 KB —
gradcam_eval_heatmaps/sample_421_true_Fetal_brain_pred_Fetal_brain.pngOther21.3 KB —
gradcam_eval_heatmaps/sample_423_true_Fetal_abdomen_pred_Other.pngOther18.9 KB —
gradcam_eval_heatmaps/sample_424_true_Fetal_femur_pred_Fetal_femur.pngOther15.8 KB —
gradcam_eval_heatmaps/sample_429_true_Fetal_brain_pred_Fetal_brain.pngOther21.4 KB —
gradcam_eval_heatmaps/sample_431_true_Fetal_femur_pred_Fetal_femur.pngOther21.3 KB —
gradcam_eval_heatmaps/sample_432_true_Fetal_brain_pred_Fetal_brain.pngOther17.7 KB —
gradcam_eval_heatmaps/sample_434_true_Fetal_femur_pred_Fetal_femur.pngOther19.8 KB —
gradcam_eval_heatmaps/sample_437_true_Fetal_brain_pred_Fetal_brain.pngOther19.4 KB —
gradcam_eval_heatmaps/sample_440_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther18.5 KB —
gradcam_eval_heatmaps/sample_440_true_Fetal_abdomen_pred_Other.pngOther19.2 KB —
gradcam_eval_heatmaps/sample_444_true_Fetal_femur_pred_Fetal_femur.pngOther19.5 KB —
gradcam_eval_heatmaps/sample_446_true_Fetal_brain_pred_Fetal_brain.pngOther18.9 KB —
gradcam_eval_heatmaps/sample_44_true_Fetal_brain_pred_Fetal_brain.pngOther20.8 KB —
gradcam_eval_heatmaps/sample_47_true_Fetal_thorax_pred_Other.pngOther19.8 KB —
gradcam_eval_heatmaps/sample_49_true_Fetal_femur_pred_Fetal_femur.pngOther17.7 KB —
gradcam_eval_heatmaps/sample_51_true_Fetal_brain_pred_Fetal_brain.pngOther18.9 KB —
gradcam_eval_heatmaps/sample_52_true_Fetal_thorax_pred_Other.pngOther17.2 KB —
gradcam_eval_heatmaps/sample_57_true_Fetal_thorax_pred_Other.pngOther19.4 KB —
gradcam_eval_heatmaps/sample_63_true_Fetal_femur_pred_Fetal_femur.pngOther25.6 KB —
gradcam_eval_heatmaps/sample_71_true_Fetal_femur_pred_Fetal_femur.pngOther21.9 KB —
gradcam_eval_heatmaps/sample_79_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther19.5 KB —
gradcam_eval_heatmaps/sample_81_true_Fetal_brain_pred_Fetal_brain.pngOther17.6 KB —
gradcam_eval_heatmaps/sample_83_true_Fetal_thorax_pred_Fetal_thorax.pngOther17.9 KB —
gradcam_eval_heatmaps/sample_87_true_Fetal_thorax_pred_Fetal_thorax.pngOther22.3 KB —
gradcam_eval_heatmaps/sample_98_true_Fetal_abdomen_pred_Fetal_abdomen.pngOther17.9 KB —
loss_curve.pngOther181.9 KB 048c9f8b74d4
training_history.csvOther3.2 KB —
validation_metrics_curve.pngOther130.5 KB 494a1976ddcd
.gitattributesRepository2.6 KB —

License and Download

License
mit
Access
Open weights, no gate
Download size
44.8 MB
Download from AI/ML Mentorship

Released by AI/ML Mentorship through its official repository on Hugging Face. Read the license.

Built From

  • Trained on (disclosed) Marc-HealthAI/fetal-planes-classification-dataset-main

Memory Requirements

PrecisionWeights in memory
As published44.8 MB
16-bit0.0 GB
8-bit0.0 GB
4-bit0.0 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About Crimson-Flame-6-resNet18

How much GPU memory does Crimson-Flame-6-resNet18 need?

About 0 GB at 16-bit and 0 GB at 4-bit: the weights (11M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run Crimson-Flame-6-resNet18 on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use Crimson-Flame-6-resNet18 commercially?

Yes. Crimson-Flame-6-resNet18 is released under MIT License. The MIT License is a short permissive license. It permits commercial use, modification and redistribution, provided the copyright notice and permission notice are included.

Similar Models

Model · Image classification

resnet18.a1_in1k

PyTorch Image Models

A ResNet-B image classification model. single layer 7x7 convolution with pooling 1x1 convolution shortcut downsample Trained on ImageNet-1k in timm using recipe template described below. ResNet Strikes Back A1 recipe - Deep Residual Learning for Image Recognition: https://arxiv.org/abs/1512.03385 Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 12M parameters timm

Model · Image classification

resnet18.a3_in1k

PyTorch Image Models

A ResNet-B image classification model. single layer 7x7 convolution with pooling 1x1 convolution shortcut downsample Trained on ImageNet-1k in timm using recipe template described below. ResNet Strikes Back A3 recipe - Deep Residual Learning for Image Recognition: https://arxiv.org/abs/1512.03385 Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 12M parameters timm

Model · Image classification

efficientnet_b3.ra2_in1k

PyTorch Image Models

A EfficientNet image classification model. Trained on ImageNet-1k in timm using recipe template described below. RandAugment RA2 recipe. Inspired by and evolved from EfficientNet RandAugment recipes. Published as B recipe in ResNet Strikes Back. RMSProp (TF 1.0 behaviour) optimizer, EMA weight averaging Step (exponential decay w/ staircase) LR schedule with warmup Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 12M parameters timm

Model · Image classification

cspnext_m.rsb_a1_in1k

Munehiro Kobayashi

A CSPNeXt image classification model. Pretrained on ImageNet-1k by OpenMMLab (RTMDet) and converted to timm format. Name disambiguation: This is the CSPNeXt backbone of RTMDet (OpenMMLab), as implemented in MMDetection. A separate paper, CSPNeXt: A new efficient token hybrid backbone (Chen et al., EAAI 2024, doi:10.1016/j.engappai.2024.107886), uses the same name for a different architecture. These weights do not implement that paper; please cite RTMDet (below) for this model. - https://github.com/open-mmlab/mmdetection/tree/main/configs/rtmdet/classification - https://download.openmmlab.com/mmdetection/v3.0/rtmdet/cspnextrsbpretrain/cspnext-m8xb256-rsb-a1-600ein1k-ecb3bbd9.pth ImageNet-1k…

Open weights apache-2.0 13M parameters timm

Model · Image classification

repvgg_a0.rvgg_in1k

PyTorch Image Models

A RepVGG image classification model. Trained on ImageNet-1k by paper authors. This model architecture is implemented using timm's flexible BYOBNet (Bring-Your-Own-Blocks Network). block / stage layout stem layout output stride (dilation) activation and norm layers channel and spatial / self-attention layers...and also includes timm features common to many other architectures, including: stochastic depth gradient checkpointing layer-wise LR decay per-stage feature extraction Explore the dataset and runtime metrics of this model in timm model results.

Open weights mit 9M parameters timm

Model · Image classification

convnextv2_pico.fcmae_ft_in1k

PyTorch Image Models

A ConvNeXt-V2 image classification model. Pretrained with a fully convolutional masked autoencoder framework (FCMAE) and fine-tuned on ImageNet-1k. Explore the dataset and runtime metrics of this model in timm model results. All timing numbers from eager model PyTorch 1.13 on RTX 3090 w/ AMP.

Open weights cc-by-nc-4.0 9M parameters timm