@digipair/skill-tensorflow 0.91.0-0 → 0.92.0

package/index.cjs.d.ts

@@ -0,0 +1 @@
+export * from "./src/index";

package/index.cjs.js

@@ -0,0 +1,65 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var tf = require('@tensorflow/tfjs-node');
+var cocoSsdModel = require('@tensorflow-models/coco-ssd');
+var faceDetectionModel = require('@tensorflow-models/face-detection');
+
+function _interopNamespace(e) {
+  if (e && e.__esModule) return e;
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n["default"] = e;
+  return Object.freeze(n);
+}
+
+var tf__namespace = /*#__PURE__*/_interopNamespace(tf);
+var cocoSsdModel__namespace = /*#__PURE__*/_interopNamespace(cocoSsdModel);
+var faceDetectionModel__namespace = /*#__PURE__*/_interopNamespace(faceDetectionModel);
+
+let TensorflowService = class TensorflowService {
+    async base64ToImage(base64) {
+        const data = base64.replace(/^data:image\/\w+;base64,/, '');
+        const buffer = Buffer.from(data, 'base64');
+        return buffer;
+    }
+    async cocoSsd(params, _pinsSettingsList, _context) {
+        const { image, base = 'lite_mobilenet_v2', modelUrl } = params;
+        const model = await cocoSsdModel__namespace.load({
+            base,
+            modelUrl
+        });
+        const imageLoaded = await this.base64ToImage(image);
+        const imageData = tf__namespace.node.decodeImage(imageLoaded, 3);
+        const predictions = await model.detect(imageData);
+        return predictions;
+    }
+    async faceDetection(params, _pinsSettingsList, _context) {
+        const { image, runtime = 'mediapipe' } = params;
+        const model = faceDetectionModel__namespace.SupportedModels.MediaPipeFaceDetector;
+        const imageLoaded = await this.base64ToImage(image);
+        const imageData = tf__namespace.node.decodeImage(imageLoaded, 3);
+        const detectorConfig = {
+            runtime
+        };
+        const detector = await faceDetectionModel__namespace.createDetector(model, detectorConfig);
+        const faces = await detector.estimateFaces(imageData);
+        return faces;
+    }
+};
+const cocoSsd = (params, pinsSettingsList, context)=>new TensorflowService().cocoSsd(params, pinsSettingsList, context);
+const faceDetection = (params, pinsSettingsList, context)=>new TensorflowService().faceDetection(params, pinsSettingsList, context);
+
+exports.cocoSsd = cocoSsd;
+exports.faceDetection = faceDetection;

package/index.esm.js

@@ -0,0 +1,38 @@
+import * as tf from '@tensorflow/tfjs-node';
+import * as cocoSsdModel from '@tensorflow-models/coco-ssd';
+import * as faceDetectionModel from '@tensorflow-models/face-detection';
+
+let TensorflowService = class TensorflowService {
+    async base64ToImage(base64) {
+        const data = base64.replace(/^data:image\/\w+;base64,/, '');
+        const buffer = Buffer.from(data, 'base64');
+        return buffer;
+    }
+    async cocoSsd(params, _pinsSettingsList, _context) {
+        const { image, base = 'lite_mobilenet_v2', modelUrl } = params;
+        const model = await cocoSsdModel.load({
+            base,
+            modelUrl
+        });
+        const imageLoaded = await this.base64ToImage(image);
+        const imageData = tf.node.decodeImage(imageLoaded, 3);
+        const predictions = await model.detect(imageData);
+        return predictions;
+    }
+    async faceDetection(params, _pinsSettingsList, _context) {
+        const { image, runtime = 'mediapipe' } = params;
+        const model = faceDetectionModel.SupportedModels.MediaPipeFaceDetector;
+        const imageLoaded = await this.base64ToImage(image);
+        const imageData = tf.node.decodeImage(imageLoaded, 3);
+        const detectorConfig = {
+            runtime
+        };
+        const detector = await faceDetectionModel.createDetector(model, detectorConfig);
+        const faces = await detector.estimateFaces(imageData);
+        return faces;
+    }
+};
+const cocoSsd = (params, pinsSettingsList, context)=>new TensorflowService().cocoSsd(params, pinsSettingsList, context);
+const faceDetection = (params, pinsSettingsList, context)=>new TensorflowService().faceDetection(params, pinsSettingsList, context);
+
+export { cocoSsd, faceDetection };

package/libs/skill-tensorflow/src/lib/skill-tensorflow.d.ts

@@ -0,0 +1,5 @@
+import { PinsSettings } from '@digipair/engine';
+import * as cocoSsdModel from '@tensorflow-models/coco-ssd';
+import * as faceDetectionModel from '@tensorflow-models/face-detection';
+export declare const cocoSsd: (params: any, pinsSettingsList: PinsSettings[], context: any) => Promise<cocoSsdModel.DetectedObject[]>;
+export declare const faceDetection: (params: any, pinsSettingsList: PinsSettings[], context: any) => Promise<faceDetectionModel.Face[]>;

package/package.json

@@ -1,28 +1,17 @@
 {
   "name": "@digipair/skill-tensorflow",
-  "version": "0.91.0-0",
-  "type": "module",
-  "main": "dist/libs/skill-tensorflow/index.cjs.js",
-  "module": "dist/libs/skill-tensorflow/index.esm.js",
-  "types": "dist/libs/skill-tensorflow/index.esm.d.ts",
-  "exports": {
-    "./package.json": "./libs/skill-tensorflow/package.json",
-    ".": {
-      "development": "./dist/libs/skill-tensorflow/src/index.ts",
-      "types": "./dist/libs/skill-tensorflow/index.esm.d.ts",
-      "import": "./dist/libs/skill-tensorflow/index.esm.js",
-      "default": "./dist/libs/skill-tensorflow/index.cjs.js"
-    }
-  },
+  "version": "0.92.0",
   "keywords": [
     "digipair",
-    "util",
-    "service"
+    "service",
+    "util"
   ],
-  "nx": {
-    "name": "skill-tensorflow"
-  },
   "dependencies": {
-    "@digipair/engine": "0.91.0-0"
-  }
-}
+    "@tensorflow-models/coco-ssd": "^2.2.3",
+    "@tensorflow-models/face-detection": "^1.0.2",
+    "@tensorflow/tfjs-node": "^4.20.0",
+    "@mediapipe/face_detection": "^0.4.1646425229"
+  },
+  "main": "./index.cjs.js",
+  "module": "./index.esm.js"
+}

package/.swcrc

@@ -1,28 +0,0 @@
-{
-  "jsc": {
-    "target": "es2017",
-    "parser": {
-      "syntax": "typescript",
-      "decorators": true,
-      "dynamicImport": true
-    },
-    "transform": {
-      "decoratorMetadata": true,
-      "legacyDecorator": true
-    },
-    "keepClassNames": true,
-    "externalHelpers": true,
-    "loose": true
-  },
-  "module": {
-    "type": "es6"
-  },
-  "sourceMaps": true,
-  "exclude": [
-    "jest.config.ts",
-    ".*\\.spec.tsx?$",
-    ".*\\.test.tsx?$",
-    "./src/jest-setup.ts$",
-    "./**/jest-setup.ts$"
-  ]
-}

package/README.md

@@ -1,7 +0,0 @@
-# mylib
-
-This library was generated with [Nx](https://nx.dev).
-
-## Building
-
-Run `nx build mylib` to build the library.

package/eslint.config.mjs

@@ -1,22 +0,0 @@
-import baseConfig from '../../eslint.config.mjs';
-
-export default [
-  ...baseConfig,
-  {
-    files: ['**/*.json'],
-    rules: {
-      '@nx/dependency-checks': [
-        'error',
-        {
-          ignoredFiles: [
-            '{projectRoot}/eslint.config.{js,cjs,mjs}',
-            '{projectRoot}/rollup.config.{js,ts,mjs,mts,cjs,cts}',
-          ],
-        },
-      ],
-    },
-    languageOptions: {
-      parser: await import('jsonc-eslint-parser'),
-    },
-  },
-];

package/rollup.config.cjs

@@ -1,28 +0,0 @@
-const { withNx } = require('@nx/rollup/with-nx');
-
-module.exports = withNx(
-  {
-    main: 'libs/skill-tensorflow/src/index.ts',
-    outputPath: 'dist/libs/skill-tensorflow',
-    tsConfig: 'libs/skill-tensorflow/tsconfig.lib.json',
-    compiler: 'swc',
-    format: ['esm', "cjs"],
-    assets: [
-      {
-        input: 'libs/skill-tensorflow/',
-        glob: 'package.json',
-        output: '.'
-      },
-      {
-        input: 'libs/skill-tensorflow/src/',
-        glob: '*.json',
-        output: '.'
-      }
-    ]
-  },
-  {
-    // Provide additional rollup configuration here. See: https://rollupjs.org/configuration-options
-    // e.g.
-    // output: { sourcemap: true },
-  }
-);

package/src/lib/skill-tensorflow.spec.ts

@@ -1,7 +0,0 @@
-import { skillTensorflow } from './skill-tensorflow';
-
-describe('skillTensorflow', () => {
-  it('should work', () => {
-    expect(skillTensorflow()).toEqual('skill-tensorflow');
-  });
-});

package/src/lib/skill-tensorflow.ts

@@ -1,48 +0,0 @@
-import { PinsSettings } from '@digipair/engine';
-import * as tf from '@tensorflow/tfjs-node';
-import * as cocoSsdModel from '@tensorflow-models/coco-ssd';
-import * as faceDetectionModel from '@tensorflow-models/face-detection';
-import { MediaPipeFaceDetectorMediaPipeModelConfig } from '@tensorflow-models/face-detection';
-
-class TensorflowService {
-  private async base64ToImage(base64: string) {
-    const data = base64.replace(/^data:image\/\w+;base64,/, '');
-    const buffer = Buffer.from(data, 'base64');
-    return buffer
-  };
-
-  async cocoSsd(params: any, _pinsSettingsList: PinsSettings[], _context: any) {
-    const { image, base = 'lite_mobilenet_v2', modelUrl } = params;
-
-    const model = await cocoSsdModel.load({
-      base,
-      modelUrl
-    });
-    const imageLoaded = await this.base64ToImage(image);
-    const imageData = tf.node.decodeImage(imageLoaded, 3);
-    const predictions = await model.detect(imageData as any);
-
-    return predictions;
-  }
-
-  async faceDetection(params: any, _pinsSettingsList: PinsSettings[], _context: any) {
-    const { image, runtime = 'mediapipe' } = params;
-
-    const model = faceDetectionModel.SupportedModels.MediaPipeFaceDetector;
-    const imageLoaded = await this.base64ToImage(image);
-    const imageData = tf.node.decodeImage(imageLoaded, 3);
-    const detectorConfig: MediaPipeFaceDetectorMediaPipeModelConfig = {
-      runtime, // 'mediapipe' or 'tfjs'
-    }
-    const detector = await faceDetectionModel.createDetector(model, detectorConfig);
-    const faces = await detector.estimateFaces(imageData as any);
-
-    return faces;
-  }
-}
-
-export const cocoSsd = (params: any, pinsSettingsList: PinsSettings[], context: any) =>
-  new TensorflowService().cocoSsd(params, pinsSettingsList, context);
-
-export const faceDetection = (params: any, pinsSettingsList: PinsSettings[], context: any) =>
-  new TensorflowService().faceDetection(params, pinsSettingsList, context);

package/tsconfig.json

@@ -1,13 +0,0 @@
-{
-  "extends": "../../tsconfig.base.json",
-  "files": [],
-  "include": [],
-  "references": [
-    {
-      "path": "../engine"
-    },
-    {
-      "path": "./tsconfig.lib.json"
-    }
-  ]
-}

package/tsconfig.lib.json

@@ -1,19 +0,0 @@
-{
-  "extends": "../../tsconfig.base.json",
-  "compilerOptions": {
-    "rootDir": "src",
-    "outDir": "dist",
-    "tsBuildInfoFile": "dist/tsconfig.lib.tsbuildinfo",
-    "emitDeclarationOnly": true,
-    "module": "esnext",
-    "moduleResolution": "node",
-    "forceConsistentCasingInFileNames": true,
-    "types": ["node"]
-  },
-  "include": ["src/**/*.ts"],
-  "references": [
-    {
-      "path": "../engine/tsconfig.lib.json"
-    }
-  ]
-}