Using 0, 1 label images instead of 0, 255