img = cv2.resize(img, dsize=(128, 128), interpolation=cv2.INTER_CUBIC)
img = self.aug(img)
img = np.transpose(img, (2,0,1)).astype(np.float32) 

pytorch transforms only works on tensors, but you apply them on numpy arrays: img = self.aug(img)