Hello, great work! If you want, you can include my opensource library for 2D/3D medical image segmentation Link: https://github.com/black0017/MedicalZooPytorch All the best in your work Adrian. N.A.