@tensorflow-models/coco-ssd
Version:
Object detection model (coco-ssd) in TensorFlow.js
20 lines (19 loc) • 764 B
TypeScript
import { tensorflow_json } from '../data/compiled_api_json';
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_json.IGraphDef): Graph;
private mapNode(node);
private decodeBase64(text);
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);
}