Onypony

Thank you for sharing the code. I believe the line (# 135 in trainer.py) when you are creating the mask is incorrect, because you use img.shape[0] (which is the number...