Hello @andrewekhalel! Sorry for taking so long to get back to you. This looks like a much better architecture diagram for the U-Net! Since this is your work and not mine, and since I'm less familiar ...