save masks in separate folder
prediction visualisation test