I’m not deeply familiar with albumentations and don’t know which internal data types are used, but based on the error message it seems you cannot transform a uint16 numpy array to a tensor so you might need to transform it to e.g. int32 first:
a = np.random.randint(0, 10, (3, 24, 24)).astype(np.uint16)
x = torch.from_numpy(a)
> TypeError: can't convert np.ndarray of type numpy.uint16. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.

x = torch.from_numpy(a.astype(np.int32))