transformers-fork/src/models/efficientnet/image_processing_efficientnet.js

State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!

import { 
    ImageProcessor,
} from "../../base/image_processors_utils.js";

export class EfficientNetImageProcessor extends ImageProcessor {
    constructor(config) {
        super(config);
        this.include_top = this.config.include_top ?? true;
        if (this.include_top) {
            this.image_std = this.image_std.map(x => x * x);
        }
    }
}