@JuanFMontesinos Thank you for the comment. Good points. Yes, I work on echocardiography images. It does have two classes. At the moment U-net is giving me reasonable accuracy. I also tried three different models, but U-net is the winner.