Mouse Brain
Segmentation automatique du cerveau de souris à partir d'images IRM via un réseau neuronal U-NET.
Données
La base de données initiale est composée de 16 images 3d labellisées de cerveaux de souris.
Nous l'augmentons en effectuant des symétries et rotation aléatoires.
Environnement python
python -m venv venv
source venv/bin/activate
pip install -r requirements.txt
Entrainement du modèle
python src/train.py
Le modèle est sauvegardé dans le fichier models/unet3d-xxxxxx.h5
Evaluation du modèle
python src/test.py
Résultats
Entrainement sur 15 epochs avec 128 données d'entrainements, et un batchsize de 4.
Modèle: models/unet3d-15812270355.h5
History: models/unet3d-15812270355-history.csv
Pour regénérer la courbe, utiliser le script python src/plot.py
.