ntedgi / node-efficientnet

tensorflowJS implementation of EfficientNet 🚀
https://www.npmjs.com/package/node-efficientnet
MIT License
257 stars 37 forks source link

Efficientnet trained with noisystudent #59

Open marcelklehr opened 3 years ago

marcelklehr commented 3 years ago

Is your feature request related to a problem? Please describe. NoisyStudent is a training method that achieves better results.

Describe the solution you'd like It would be cool if you could also offer NoisyStudent variants of EfficientNet

ntedgi commented 3 years ago

thanks, @marcelklehr for open this issue

try follow this repo https://github.com/ntedgi/efficientnet-tensorflowjs-binaries

it contains node-efficientnet tensor flow model there is a simple example on how to transfer tf-model after you finish transfer you can load it here in the same model

marcelklehr commented 3 years ago

Would you be willing to accept a PR that adds NoisyStudent checkpoints?

ntedgi commented 3 years ago

Definitely and if you need help I would be happy to help

ntedgi commented 3 years ago

Before you start modifying the code base try to load your model with this example

‘const path = require("path"); const { EfficientNetCheckPointFactory, EfficientNetCheckPoint } = require("node-efficientnet");

const model = await EfficientNetCheckPointFactory.create( EfficientNetCheckPoint.B7, { localModelRootDirectory: path.join(__dirname, "local_model") } );

const path2image = "...";

const topResults = 5;

const result = await model.inference(path2image, { topK: topResults, locale: "zh" });’