tensorflow-helpers/image-model.d.ts

Helper functions to use tensorflow in nodejs for transfer learning, image classification, and more

export type ImageModelSpec = {
    url: string;
    width: number;
    height: number;
    channels: number;
    features: number;
};
export declare const PreTrainedImageModels: {
    mobilenet: {
        'mobilenet-v3-large-100': {
            url: "https://www.kaggle.com/models/google/mobilenet-v3/TfJs/large-100-224-feature-vector/1";
            width: 224;
            height: 224;
            channels: 3;
            features: 1280;
        };
        'mobilenet-v3-large-75': {
            url: "https://www.kaggle.com/models/google/mobilenet-v3/TfJs/large-075-224-feature-vector/1";
            width: 224;
            height: 224;
            channels: 3;
            features: 1280;
        };
        'mobilenet-v3-small-100': {
            url: "https://www.kaggle.com/models/google/mobilenet-v3/TfJs/small-100-224-feature-vector/1";
            width: 224;
            height: 224;
            channels: 3;
            features: 1280;
        };
        'mobilenet-v3-small-75': {
            url: "https://www.kaggle.com/models/google/mobilenet-v3/TfJs/small-075-224-feature-vector/1";
            width: 224;
            height: 224;
            channels: 3;
            features: 1280;
        };
    };
};