I am trying to use a UNet for biomedical image segmentation with a Kaggle lung dataset. For the larger organs such as the Lungs and Liver my UNet works good but for other or