search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1.01k stars 330 forks source link

Add EfficientNet Lite model variants. #630

Closed sebastian-sz closed 1 year ago

sebastian-sz commented 2 years ago

Short Description EfficientNet Lite models are lighter variants of EfficientNets more suitable for mobile deployment.

Existing Implementations Original implementation - link My reimplementation - link

Other Information EfficientNet Lite's have been requested multiple times in the Tensorflow/Keras ecosystem:

https://github.com/keras-team/tf-keras/issues/502 https://github.com/tensorflow/tensorflow/issues/45091 https://github.com/tensorflow/tensorflow/pull/48400

Let me know what you think!

MrinalTyagi commented 2 years ago

Would love to pick this up if noone else is currently working on this. @sebastian-sz

sebastian-sz commented 2 years ago

@MrinalTyagi I actually wanted to add these models, but I'm waiting for:

  1. Acceptance / contribution welcome tag from the repo authors.
  2. When keras-team/keras#570 is merged

There should also be some API discussion: do we want these as separate models, like EfficientNetLiteB0 or kind of like MobilenetV3 add a minimalistic or lite flag to existing implementation.