Spaces:
Running
Running
convert-gemma3-to-onnx / transformers.js /src /models /efficientnet /image_processing_efficientnet.js
| import { | |
| ImageProcessor, | |
| } from "../../base/image_processors_utils.js"; | |
| export class EfficientNetImageProcessor extends ImageProcessor { | |
| constructor(config) { | |
| super(config); | |
| // @ts-expect-error TS2339 | |
| this.include_top = this.config.include_top ?? true; | |
| if (this.include_top) { | |
| this.image_std = this.image_std.map(x => x * x); | |
| } | |
| } | |
| } | |