-
Notifications
You must be signed in to change notification settings - Fork 26
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: use MobileNetV4 Core ML model with NCHW input #895
Conversation
MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅ |
There are 2 issues:
|
@freedomtan to check the NHWC -> NCHW conversion. |
@anhappdev this NHWC -> NCHW looks right to me. Maybe you can try to compare with the result of using a single transpose op TFLite model (Yes, it's posssible. E.g., in TFLite label_image, I constructed a TFLite model with only |
|
There was an issue with wrong data type, but I got it working now. |
The models can be downloaded from here:
https://drive.google.com/drive/folders/1-Mloub0e41mkYs09i9tQZl95dTnGdZ6f?usp=sharing