I have made some 2D numpy arrays (values between 0~1 , black and white, 1 channel) that I would like to train them as images for classification, I was wondering if it's possible to pass the arrays to ...

where last hist_i input takes 2D array in original model. But MLFlow reads this inputs and converts it into a pandas dataframe where each column is an input. So here, one of the column would have ...