UNPKG

@tensorflow-models/coco-ssd

Version:

Object detection model (coco-ssd) in TensorFlow.js

19 lines (18 loc) 716 B
import { tensorflow } from '../data/compiled_api'; import { Graph } from './types'; export declare class OperationMapper { private static _instance; private opMappers; static readonly Instance: OperationMapper; private constructor(); private isControlFlow(node); private isDynamicShape(node); transformGraph(graph: tensorflow.IGraphDef): Graph; private mapNode(node); private getStringParam(attrs, name, def, keepCase?); private getBoolParam(attrs, name, def); private getNumberParam(attrs, name, def); private getDtypeParam(attrs, name, def); private getTensorShapeParam(attrs, name, def?); private getNumericArrayParam(attrs, name, def); }