budai4medtech / xfetus

xfetus -- [x]ynthetic[fetus] (:baby: :robot:) -- A library for ultrasound fetal imaging synthesis using techniques from GANs, transformers, and diffusion models.
5 stars 1 forks source link

Implementation of classifier ResNet18 #37

Open mxochicale opened 11 months ago

mxochicale commented 11 months ago

Raising this one to implement ResNet18 classifier based on the work of Montero et al. 2021. See "Supplementary Materials The following are available at https://www.mdpi.com/article/10.3390/s21237975/s1, Section S1: ResNet18 Fastai Architecture" : https://www.mdpi.com/1424-8220/21/23/7975

Description of Maternal fetal ultrasound planes from low-resource imaging settings in five African countries https://github.com/budai4medtech/xfetus/tree/main/data

Other relevant papers:

mxochicale commented 11 months ago

Thanks @michellepi for pointed out the fetal plane classifier from Sendra-Balcells et al. 2023. Sounds like a great idea to have a classifier that can be used for different datasets (e.g. Spain common maternal-fetal, African countries datasets and our synthesised datasets). Feel free to make contributions.

Based on the model benchmark presented by Burgos et al.18 for fetal plane classification, the best performing method, a Densenet-16925, was implemented in this study.

Sendra-Balcells, Carla, Víctor M. Campello, Jordina Torrents-Barrena, Yahya Ali Ahmed, Mustafa Elattar, Benard Ohene-Botwe, Pempho Nyangulu et al. "Generalisability of fetal ultrasound deep learning models to low-resource imaging settings in five African countries." Scientific reports 13, no. 1 (2023): 2728.

If we go for config files, these are few pointers (but might be better options):