image-dataset/dist/cache.js

Tool to build image dataset: collect, classify, review

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.datasetCache = exports.modelsCache = void 0;
exports.makeCache = makeCache;
const model_1 = require("./model");
function makeCache(fn, clearFn) {
    let value = null;
    async function load() {
        if (value != null && clearFn) {
            await clearFn(value);
        }
        value = await fn();
    }
    async function get() {
        value ||= await fn();
        return value;
    }
    return { load, get };
}
exports.modelsCache = makeCache(() => (0, model_1.loadModels)());
exports.datasetCache = makeCache(async () => {
    let models = await exports.modelsCache.get();
    return models.classifierModel.loadDatasetFromDirectory();
});