ntedgi/node-efficientnet

Efficientnet trained with noisystudent

Opened this issue · 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

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

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

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

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