I think that mask should be renamed to mask_img. I think that a consistent naming convention to see types quickly will help usability.