def inverse(self, a: np.array):
last_dim = len(a.shape) - 1
for t in self.tfms:
if isinstance(t, A.HorizontalFlip):
a = flip(a, last_dim)
elif isinstance(t, A.VerticalFlip):
a = flip(a, last_dim - 1)
elif isinstance(t, A.Transpose):
axis = (0, 1, 3, 2) if len(a.shape) == 4 else (0, 2, 1)
a = a.permute(*axis)
return a
def to_rgb(batch: np.array) -> np.array:
batch = batch.permute(0, 2, 3, 1).numpy() if len(
batch.shape) == 4 else batch.permute(0, 2, 3, 4, 1).numpy()
batch = (batch + 1) / 2
batch = (batch * 255).astype(np.uint8)
return batch
def to_rgb(batch: np.array) -> np.array:
batch = (batch.permute(0, 2, 3, 1).numpy() + 1) / 2
batch = (batch * 255).astype(np.uint8)
return batch
