Skip to content

Add MobileNetV2 0.5 Plant Village classifier#25

Open
nikunj95 wants to merge 5 commits into
mainfrom
add-mobilenetv2-plant-village
Open

Add MobileNetV2 0.5 Plant Village classifier#25
nikunj95 wants to merge 5 commits into
mainfrom
add-mobilenetv2-plant-village

Conversation

@nikunj95
Copy link
Copy Markdown

Summary

  • New vision/classification/mobilenetv2/plant_village/ folder with float and 8w8a8 QAT Keras checkpoints, tracked via Git LFS.
  • Model card covers environment pins (akida==2.17.0, cnn2snn==2.17.0, quantizeml==1.0.1), performance, and a load-and-convert-to-Akida snippet ported from the reference notebook.
  • Added the Plant Village row to the top-level model table in README.md.

Model

  • Architecture: MobileNetV2, alpha = 0.5, 224 × 224 × 3 input, 38 classes
  • Backbone init: ImageNet pretrained
  • Quantization: 8-bit weights / 8-bit activations (QAT)
  • Minimal Akida nodes: 6 (akida.SixNodesIPv2, hw_only=True)
Variant Top-1 accuracy
Float 99.47%
Quantized 8w8a8 (QAT) 99.12%

Quantized-model sparsity — ReLU: 89%, overall: 53%.

- Add float and 8w8a8 QAT Keras checkpoints under
  vision/classification/mobilenetv2/plant_village/ (tracked via Git LFS).
- Add model README with environment pins and a load+convert-to-Akida snippet
  taken from the reference notebook.
- Add Plant Village row to the top-level model table.
Comment thread vision/classification/mobilenetv2/plant_village/README.md Outdated
Comment thread vision/classification/mobilenetv2/plant_village/README.md
Comment thread vision/classification/mobilenetv2/plant_village/README.md Outdated
Comment thread README.md Outdated
| Vision | Classification | [MobileNetV2 1.0](vision/classification/mobilenetv2/imagenet) | 224 | ImageNet | 3.5M | 8 | 70.35% | | | 7 |
| Vision | Classification | [MobileNetV2 0.75](vision/classification/mobilenetv2/imagenet) | 160 | ImageNet | 2.6M | 8 | 62.85% | | | 4 ☁️ |
| Vision | Classification | [MobileNetV2 0.35](vision/classification/mobilenetv2/imagenet) | 96 | ImageNet | 1.2M | 8 | 43.47% | | | 2 ☁️ |
| Vision | Classification | [MobileNetV2 0.5](vision/classification/mobilenetv2/plant_village) | 224 | Plant Village | 0.7M | 8 | 99.12% | | | 6 |
Copy link
Copy Markdown
Collaborator

@ktsiknos-brainchip ktsiknos-brainchip Apr 21, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Manually tested using the CI script: this model can map on 4 nodes, not 6.

Also, add the ☁️ icon if this maps on 6-nodes (it just means it could be run on the cloud that has 6-nodes)

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I confirmed it on my side too, using MapMode value 3.
cc. @kmanninen

Comment thread vision/classification/mobilenetv2/plant_village/README.md
Comment thread vision/classification/mobilenetv2/plant_village/README.md Outdated
Copy link
Copy Markdown

@kmanninen kmanninen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Recommend implementing changes pertaining to using the latest version of Akida.

kmanninen and others added 2 commits April 21, 2026 15:15
Co-authored-by: Kevin Tsiknos <60514526+ktsiknos-brainchip@users.noreply.github.com>
Comment thread vision/classification/mobilenetv2/plant_village/README.md Outdated
Comment thread README.md Outdated
Copy link
Copy Markdown

@kmanninen kmanninen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I approve.

@ktsiknos-brainchip
Copy link
Copy Markdown
Collaborator

ktsiknos-brainchip commented Apr 22, 2026

@nikunj95 @kmanninen few comments:

  • this needs rebased on main branch
  • it would be best to squash the suggested changes commit under a 'review' commit to preserve a clean history
  • the README line for the new model is still missing the ☁️ icon

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants