I noticed that the data loading, visualization, dataset and dataloader sections seem to be based on a Kaggle notebook (link). However, some important topics, such as the explanation of the UNET model and the training, evaluation and inference loops, are not detailed.
If it is OK with you, I could propose an improved version that includes additional documentation and references to tutorials that could be useful to develop the notebook more completely.
I noticed that the data loading, visualization, dataset and dataloader sections seem to be based on a Kaggle notebook (link). However, some important topics, such as the explanation of the UNET model and the training, evaluation and inference loops, are not detailed.
If it is OK with you, I could propose an improved version that includes additional documentation and references to tutorials that could be useful to develop the notebook more completely.