

cppe5["train"] = cppe5["train"].with_transform(transform_aug_ann)
cppe5["train"][15]
{'pixel_values': tensor([[[ 0.9132,  0.9132,  0.9132,  , -1.9809, -1.9809, -1.9809],
          [ 0.9132,  0.9132,  0.9132,  , -1.9809, -1.9809, -1.9809],
          [ 0.9132,  0.9132,  0.9132,  , -1.9638, -1.9638, -1.9638],
          ,
          [-1.5699, -1.5699, -1.5699,  , -1.9980, -1.9980, -1.9980],
          [-1.5528, -1.5528, -1.5528,  , -1.9980, -1.9809, -1.9809],
          [-1.5528, -1.5528, -1.5528,  , -1.9980, -1.9809, -1.9809]],