Add the ImageClassificationPipeline #11598
This PR adds the
Since I encountered an issue with the
Please let me know if you would like me to split this PR into multiple PRs for simpler reviews, happy to do so.
@NielsRogge, happy to have your review.
Narsil left a comment
Biggest change is the
Also the tests are a bit bloated IMO compared to what they could be. In that instance, because we can't test parity with TF, we should just make the tests much simpler.
The image loading logic should be fixed too.
The rest are NITs.