mask is not being reset after new image is chosen.

[trawler0]

I plotted the mask with matplotlib each time the drag_gan method is being called. If I don't adjust it manually after I change the image, the same mask is being used still.