learning_model 1.0.51 → 1.0.52

package/dist/lib/learning/mobilenet.d.ts

@@ -1,52 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import '@tensorflow/tfjs-backend-wasm';
-import { io } from '@tensorflow/tfjs-core';
-import LearningInterface from './base';
-declare class LearningMobilenet implements LearningInterface {
-    model: tf.LayersModel | null;
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    validateRate: number;
-    isRunning: boolean;
-    isReady: boolean;
-    isTrainedDone: boolean;
-    limitSize: number;
-    mobilenetModule: tf.LayersModel | null;
-    imageExamples: Float32Array[][];
-    classNumber: string[];
-    readonly MOBILE_NET_INPUT_WIDTH = 224;
-    readonly MOBILE_NET_INPUT_HEIGHT = 224;
-    readonly MOBILE_NET_INPUT_CHANNEL = 3;
-    readonly IMAGE_NORMALIZATION_FACTOR = 255;
-    constructor({ epochs, batchSize, limitSize, learningRate, validateRate, }?: {
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-        validateRate?: number;
-    });
-    onProgress: (progress: number) => void;
-    onLoss: (loss: number) => void;
-    onEvents: (logs: any) => void;
-    onTrainBegin: (log: any) => void;
-    onTrainEnd: (log: any) => void;
-    onEpochEnd: (epoch: number, logs: any) => void;
-    load({ jsonURL, labels }: {
-        jsonURL: string;
-        labels: Array<string>;
-    }): Promise<void>;
-    private registerClassNumber;
-    private _convertToTfDataset;
-    addData(label: string, data: any): Promise<void>;
-    init(): Promise<void>;
-    private setupBackend;
-    private checkWasmSupport;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<Map<string, number>>;
-    saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void>;
-    running(): boolean;
-    ready(): boolean;
-    private _createModel;
-}
-export default LearningMobilenet;

package/dist/lib/learning/mobilenet.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/lib/utils/canvas.d.ts

@@ -1,3 +0,0 @@
-type Drawable = ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement;
-export declare function cropTo(image: Drawable, size: number, flipped?: boolean, canvas?: HTMLCanvasElement): HTMLCanvasElement;
-export {};

package/dist/lib/utils/data_manager.d.ts

@@ -1,15 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-declare class DataManager {
-    private labelMap;
-    private labelIndex;
-    private data;
-    addData(label: string, values: number[]): Promise<void>;
-    convertToTensors(): {
-        xs: tf.Tensor2D;
-        ys: tf.Tensor1D;
-    };
-    getLabelMap(): {
-        [key: string]: number;
-    };
-}
-export default DataManager;

package/dist/lib/utils/dataset.d.ts

@@ -1,6 +0,0 @@
-export interface Sample {
-    data: Float32Array;
-    label: number[];
-}
-export declare function flatOneHot(label: number, numClasses: number): number[];
-export declare function fisherYates(array: Float32Array[] | Sample[]): Float32Array[] | Sample[];

package/dist/lib/utils/tf.d.ts

@@ -1,7 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-export declare function isTensor(c: any): c is tf.Tensor;
-export declare function loadModel(): Promise<tf.LayersModel>;
-export declare function mobileNetURL(version: number): string;
-export declare function imageToTensor(data: any): tf.Tensor3D;
-export declare function capture(rasterElement: HTMLImageElement | HTMLVideoElement | HTMLCanvasElement, grayscale?: boolean): Promise<tf.Tensor<tf.Rank>>;
-export declare function cropTensor(img: tf.Tensor3D, grayscaleModel?: boolean, grayscaleInput?: boolean): tf.Tensor3D;

package/dist/utils/canvas.d.ts

@@ -1,3 +0,0 @@
-type Drawable = ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement;
-export declare function cropTo(image: Drawable, size: number, flipped?: boolean, canvas?: HTMLCanvasElement): HTMLCanvasElement;
-export {};

package/dist/utils/canvas.js

@@ -1,47 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.cropTo = void 0;
-const newCanvas = () => document.createElement('canvas');
-function cropTo(image, size, flipped = false, canvas = newCanvas()) {
-    let width;
-    let height;
-    // If ImageData
-    if (image instanceof ImageData) {
-        width = image.width;
-        height = image.height;
-    }
-    // If image, bitmap, or canvas
-    else if (image instanceof HTMLImageElement || image instanceof HTMLCanvasElement) {
-        width = image.width;
-        height = image.height;
-    }
-    // If video element
-    else if (image instanceof HTMLVideoElement) {
-        width = image.videoWidth;
-        height = image.videoHeight;
-    }
-    else {
-        throw new Error("Unsupported Drawable type");
-    }
-    const min = Math.min(width, height);
-    const scale = size / min;
-    const scaledW = Math.ceil(width * scale);
-    const scaledH = Math.ceil(height * scale);
-    const dx = scaledW - size;
-    const dy = scaledH - size;
-    canvas.width = canvas.height = size;
-    const ctx = canvas.getContext('2d');
-    // Handle ImageData separately
-    if (image instanceof ImageData) {
-        ctx.putImageData(image, ~~(dx / 2) * -1, ~~(dy / 2) * -1); // Adjust this if needed
-    }
-    else {
-        ctx.drawImage(image, ~~(dx / 2) * -1, ~~(dy / 2) * -1, scaledW, scaledH);
-    }
-    if (flipped) {
-        ctx.scale(-1, 1);
-        ctx.drawImage(canvas, size * -1, 0);
-    }
-    return canvas;
-}
-exports.cropTo = cropTo;

package/dist/utils/data_manager.d.ts

@@ -1,15 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-declare class DataManager {
-    private labelMap;
-    private labelIndex;
-    private data;
-    addData(label: string, values: number[]): Promise<void>;
-    convertToTensors(): {
-        xs: tf.Tensor2D;
-        ys: tf.Tensor1D;
-    };
-    getLabelMap(): {
-        [key: string]: number;
-    };
-}
-export default DataManager;

package/dist/utils/data_manager.js

@@ -1,62 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const tf = __importStar(require("@tensorflow/tfjs"));
-class DataManager {
-    constructor() {
-        this.labelMap = {};
-        this.labelIndex = 0;
-        this.data = [];
-    }
-    addData(label, values) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!(label in this.labelMap)) {
-                this.labelMap[label] = this.labelIndex++;
-            }
-            const numericLabel = this.labelMap[label];
-            this.data.push({ label: numericLabel, values });
-        });
-    }
-    convertToTensors() {
-        const xs = this.data.map(d => d.values);
-        const ys = this.data.map(d => d.label);
-        const xsTensor = tf.tensor2d(xs, [xs.length, xs[0].length], 'float32');
-        const ysTensor = tf.tensor1d(ys, 'float32');
-        return { xs: xsTensor, ys: ysTensor };
-    }
-    getLabelMap() {
-        return this.labelMap;
-    }
-}
-exports.default = DataManager;

package/dist/utils/dataset.d.ts

@@ -1,6 +0,0 @@
-export interface Sample {
-    data: Float32Array;
-    label: number[];
-}
-export declare function flatOneHot(label: number, numClasses: number): number[];
-export declare function fisherYates(array: Float32Array[] | Sample[]): Float32Array[] | Sample[];

package/dist/utils/dataset.js

@@ -1,21 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.fisherYates = exports.flatOneHot = void 0;
-function flatOneHot(label, numClasses) {
-    const labelOneHot = new Array(numClasses).fill(0);
-    labelOneHot[label] = 1;
-    return labelOneHot;
-}
-exports.flatOneHot = flatOneHot;
-function fisherYates(array) {
-    const length = array.length;
-    // need to clone array or we'd be editing original as we goo
-    const shuffled = array.slice();
-    for (let i = (length - 1); i > 0; i -= 1) {
-        let randomIndex;
-        randomIndex = Math.floor(Math.random() * (i + 1));
-        [shuffled[i], shuffled[randomIndex]] = [shuffled[randomIndex], shuffled[i]];
-    }
-    return shuffled;
-}
-exports.fisherYates = fisherYates;

package/dist/utils/tf.d.ts

@@ -1,7 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-export declare function isTensor(c: any): c is tf.Tensor;
-export declare function loadModel(): Promise<tf.LayersModel>;
-export declare function mobileNetURL(version: number): string;
-export declare function imageToTensor(data: any): tf.Tensor3D;
-export declare function capture(rasterElement: HTMLImageElement | HTMLVideoElement | HTMLCanvasElement, grayscale?: boolean): Promise<tf.Tensor<tf.Rank>>;
-export declare function cropTensor(img: tf.Tensor3D, grayscaleModel?: boolean, grayscaleInput?: boolean): tf.Tensor3D;

package/dist/utils/tf.js

@@ -1,126 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.cropTensor = exports.capture = exports.imageToTensor = exports.mobileNetURL = exports.loadModel = exports.isTensor = void 0;
-const tf = __importStar(require("@tensorflow/tfjs"));
-function isTensor(c) {
-    return typeof c.dataId === 'object' && typeof c.shape === 'object';
-}
-exports.isTensor = isTensor;
-function loadModel() {
-    return __awaiter(this, void 0, void 0, function* () {
-        const trainLayerV1 = 'conv_pw_13_relu';
-        const trainLayerV2 = 'out_relu';
-        var mobileNetVersion = 2;
-        const modelURL = mobileNetURL(mobileNetVersion);
-        const load_model = yield tf.loadLayersModel(modelURL);
-        if (mobileNetVersion == 1) {
-            const layer = load_model.getLayer(trainLayerV1);
-            const truncatedModel = tf.model({
-                inputs: load_model.inputs,
-                outputs: layer.output
-            });
-            const model = tf.sequential();
-            model.add(truncatedModel);
-            model.add(tf.layers.flatten());
-            return model;
-        }
-        else {
-            const layer = load_model.getLayer(trainLayerV2);
-            const truncatedModel = tf.model({
-                inputs: load_model.inputs,
-                outputs: layer.output
-            });
-            const model = tf.sequential();
-            model.add(truncatedModel);
-            model.add(tf.layers.globalAveragePooling2d({})); // go from shape [7, 7, 1280] to [1280]
-            return model;
-        }
-    });
-}
-exports.loadModel = loadModel;
-function mobileNetURL(version) {
-    if (version == 1) {
-        return "https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_1.0_224/model.json";
-    }
-    return "https://storage.googleapis.com/teachable-machine-models/mobilenet_v2_weights_tf_dim_ordering_tf_kernels_1.0_224_no_top/model.json";
-}
-exports.mobileNetURL = mobileNetURL;
-function imageToTensor(data) {
-    let tensor;
-    if (data instanceof tf.Tensor) {
-        tensor = data;
-    }
-    else {
-        // MobileNet 모델 로드
-        tensor = tf.browser.fromPixels(data);
-    }
-    return tensor;
-}
-exports.imageToTensor = imageToTensor;
-function capture(rasterElement, grayscale) {
-    return __awaiter(this, void 0, void 0, function* () {
-        return tf.tidy(() => {
-            const pixels = tf.browser.fromPixels(rasterElement);
-            // crop the image so we're using the center square
-            const cropped = cropTensor(pixels, grayscale);
-            // Expand the outer most dimension so we have a batch size of 1
-            const batchedImage = cropped.expandDims(0);
-            // Normalize the image between -1 and a1. The image comes in between 0-255
-            // so we divide by 127 and subtract 1.
-            return batchedImage.toFloat().div(tf.scalar(127)).sub(tf.scalar(1));
-        });
-    });
-}
-exports.capture = capture;
-function cropTensor(img, grayscaleModel, grayscaleInput) {
-    const size = Math.min(img.shape[0], img.shape[1]);
-    const centerHeight = img.shape[0] / 2;
-    const beginHeight = centerHeight - (size / 2);
-    const centerWidth = img.shape[1] / 2;
-    const beginWidth = centerWidth - (size / 2);
-    if (grayscaleModel && !grayscaleInput) {
-        //cropped rgb data
-        let grayscale_cropped = img.slice([beginHeight, beginWidth, 0], [size, size, 3]);
-        grayscale_cropped = grayscale_cropped.reshape([size * size, 1, 3]);
-        const rgb_weights = [0.2989, 0.5870, 0.1140];
-        grayscale_cropped = tf.mul(grayscale_cropped, rgb_weights);
-        grayscale_cropped = grayscale_cropped.reshape([size, size, 3]);
-        grayscale_cropped = tf.sum(grayscale_cropped, -1);
-        grayscale_cropped = tf.expandDims(grayscale_cropped, -1);
-        return grayscale_cropped;
-    }
-    return img.slice([beginHeight, beginWidth, 0], [size, size, 3]);
-}
-exports.cropTensor = cropTensor;