ImageClassifierData.from_arrays with augmentation

Hi all,

I tried to search the forum but still can’t get a definite answer.
If i tried to load the data from a csv (where each column is the pixel value) using ImageClassifierData.from_arrays(), can I applied data augmentation?
When I tried to apply augmentation, it throws me exception
operands could not be broadcast together with shapes (28,28,28) (3,)

Thank you