[WIP until the conversion issue is resolved] Add EfficientNet #238
* upstream/master: Update posenet lock file. (tensorflow#244) PoseNet - Split pose estimation methods into estimateSinglePose and estimateMultiplePoses, and move inputResolution specification from loading to inference time. (tensorflow#242) update peer deps for toxicity model and increase the minor version (tensorflow#243) update models peerDependencies to be individual npm instead of union (tensorflow#237) [speech-commands] Bump version to 0.3.9 (tensorflow#240) [speech-commands] Enable creating recognizer from artifacts (tensorflow#226) Update README.md (tensorflow#239) export Mobilenet as interface (tensorflow#236) skip unneeded operation for posenet single-pose allow user to provide a custom url for the mobilenet model (tensorflow#235)
@sdll A few quick questions. Apart from the conversion issues (I'll try to reproduce that locally when I get a chance), I did have some issues running the build (incorrect reference to preamble) and running the demos (no keys in efficientnets variable thus non of the buttons work). I did hack it in, but didn't get a meaningful prediction. Are those parts still work in progress?
At a higher level I was wondering how big the weights are (in MB)? And what kind of inference times you are getting?
Sorry, @tafsiri, for the confusion, I glossed over changes I had to do following those based on @manrajgrover’s suggestions. This is fixed now, I have successfully published the module locally and tested each of the variants. Here is the screenshot:
Let me know if there are any issues other than those outlined here.
Models quantized to 2 bytes weigh from 10.2 MB (
With regards to performance,
I will refactor the code to follow the factory pattern of loading the model, taking inspiration from your review of DeepLab.
Thanks for your time and help!
@tafsiri, a fresh install did not show any issues:
# cd tfjs-models/efficientnet # git pull cd demo rm -rf node_modules/ .yalc/ .cache/ cd .. yarn publish-local cd demo yarn yarn link-local yarn watch
One thing that I did notice though was that labels were off in Chrome due to the differences in CSS rendering:
This is fixed now: