Build and train the U-Net model for segmentation
Save the trained model and training history
Files to Complete:
src/train_unet.py
src/unet_model.py (if custom model architecture is required)
Task Instructions:
Load the preprocessed data and generated segmentation masks
Train the U-Net model for the segmentation task
Save the trained model to models/unet/unet_model.keras
Save the training history and plot accuracy and loss charts
This task can begin after the data preprocessing and mask generation tasks are completed.
Task Details:
Build and train the U-Net model for segmentation Save the trained model and training history Files to Complete:
src/train_unet.py src/unet_model.py (if custom model architecture is required) Task Instructions:
Load the preprocessed data and generated segmentation masks Train the U-Net model for the segmentation task Save the trained model to models/unet/unet_model.keras Save the training history and plot accuracy and loss charts
This task can begin after the data preprocessing and mask generation tasks are completed.