Todo.
- Preprocessing: based on the provided annotations, come up with binary mask for each image.
- Model: implement in PyTorch a U-Net model for segmentation.
- Hyper-parameters: add hydra for hyper-parameters management.
- Tracking: add wandb for tracking experiments.