I am working on ML task where is one of core sub tasks is the image classification. The pre-trained model as well as normalization tool has requirements on data input format