node-efficientnet icon indicating copy to clipboard operation
node-efficientnet copied to clipboard

Efficientnet trained with noisystudent

Open marcelklehr opened this issue 4 years ago β€’ 4 comments

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

marcelklehr avatar Jul 28 '21 13:07 marcelklehr

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

ntedgi avatar Jul 28 '21 13:07 ntedgi

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

marcelklehr avatar Jul 29 '21 16:07 marcelklehr

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

ntedgi avatar Jul 29 '21 17:07 ntedgi

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" });’

ntedgi avatar Jul 30 '21 06:07 ntedgi