def _mask_transform(self, mask):
    target = (np.array(mask).astype('int64') - 1)
    return torch.from_numpy(target)
def _mask_transform(self, mask):
    target = (np.array(mask).astype('int64') - 1)
    return torch.from_numpy(target)