image-dataset
Version:
Tool to build image dataset: collect, classify, review
25 lines (24 loc) • 825 B
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const browser_1 = require("tensorflow-helpers/browser");
async function loadModels(args) {
let baseModel = await (0, browser_1.loadImageModel)({
url: '/saved_models/base_model',
cacheUrl: 'indexeddb://base_model',
});
browser_1.tf.browser.toPixels;
let classifierModel = await (0, browser_1.loadImageClassifierModel)({
baseModel,
modelUrl: '/saved_models/classifier_model',
cacheUrl: 'indexeddb://classifier_model',
checkForUpdates: true,
classNames: args.classNames,
});
return classifierModel;
}
Object.assign(window, {
loadModels,
tf: browser_1.tf,
cropAndResizeImageTensor: browser_1.cropAndResizeImageTensor,
toTensor3D: browser_1.toTensor3D,
});