tensorflow-helpers/image-model.js

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

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.PreTrainedImageModels = void 0;
exports.PreTrainedImageModels = {
    mobilenet: {
        // #param, accuracy, and latency see: https://keras.io/api/applications/mobilenet/#mobilenetv3large-function
        '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,
        },
    },
};