Preprocess for classification model
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 9.8k
- Forks
- 1.6k
- PR merge metrics
- No merged PRs in 30d
Description
Hi, I try to infer the classification models on onnxruntime engin. however, I can`t get the consistent accuracy in the page except resnet mobilenet and shufflenet. I think the preprocess of these models is different and i use the preprocess in imagenet_preprocess.py. mobilenet and shufflenet converted from pytorch got normal accuracy while inceptionv1 and efficientnet-lite4 converted from caffe got low accuracy. I guess the preprocess of the models from caffe is different.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the reported accuracy differences with the classification models in ONNX Runtime, then inspect imagenet_preprocess.py and compare its use for ResNet, MobileNet, ShuffleNet, InceptionV1, and EfficientNet-Lite4. Done means identifying whether the Caffe-converted models require different preprocessing and documenting or correcting the preprocessing needed for consistent accuracy.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100