UNPKG

@tensorflow-models/coco-ssd

Version:

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

536 lines (405 loc) 13 kB
/** Namespace tensorflow_json. */ export namespace tensorflow_json { /** Properties of an Any. */ export interface IAny { /** Any typeUrl */ typeUrl?: (string|null); /** Any value */ value?: (Uint8Array|null); } /** DataType enum. */ export enum DataType { DT_INVALID = 0, DT_FLOAT = 1, DT_DOUBLE = 2, DT_INT32 = 3, DT_UINT8 = 4, DT_INT16 = 5, DT_INT8 = 6, DT_STRING = 7, DT_COMPLEX64 = 8, DT_INT64 = 9, DT_BOOL = 10, DT_QINT8 = 11, DT_QUINT8 = 12, DT_QINT32 = 13, DT_BFLOAT16 = 14, DT_FLOAT_REF = 101, DT_DOUBLE_REF = 102, DT_INT32_REF = 103, DT_UINT8_REF = 104, DT_INT16_REF = 105, DT_INT8_REF = 106, DT_STRING_REF = 107, DT_COMPLEX64_REF = 108, DT_INT64_REF = 109, DT_BOOL_REF = 110, DT_QINT8_REF = 111, DT_QUINT8_REF = 112, DT_QINT32_REF = 113, DT_BFLOAT16_REF = 114 } /** Properties of a TensorShape. */ export interface ITensorShape { /** TensorShape dim */ dim?: (tensorflow_json.TensorShape.IDim[]|null); /** TensorShape unknownRank */ unknownRank?: (boolean|null); } export namespace TensorShape { /** Properties of a Dim. */ export interface IDim { /** Dim size */ size?: (number|string|null); /** Dim name */ name?: (string|null); } } /** Properties of a Tensor. */ export interface ITensor { /** Tensor dtype */ dtype?: (tensorflow_json.DataType|null); /** Tensor tensorShape */ tensorShape?: (tensorflow_json.ITensorShape|null); /** Tensor versionNumber */ versionNumber?: (number|null); /** Tensor tensorContent */ tensorContent?: (Uint8Array|null); /** Tensor floatVal */ floatVal?: (number[]|null); /** Tensor doubleVal */ doubleVal?: (number[]|null); /** Tensor intVal */ intVal?: (number[]|null); /** Tensor stringVal */ stringVal?: (Uint8Array[]|null); /** Tensor scomplexVal */ scomplexVal?: (number[]|null); /** Tensor int64Val */ int64Val?: ((number | string)[]|null); /** Tensor boolVal */ boolVal?: (boolean[]|null); /** Tensor uint32Val */ uint32Val?: (number[]|null); /** Tensor uint64Val */ uint64Val?: ((number | string)[]|null); } /** Properties of an AttrValue. */ export interface IAttrValue { /** AttrValue list */ list?: (tensorflow_json.AttrValue.IListValue|null); /** AttrValue s */ s?: (string|null); /** AttrValue i */ i?: (number|string|null); /** AttrValue f */ f?: (number|null); /** AttrValue b */ b?: (boolean|null); /** AttrValue type */ type?: (tensorflow_json.DataType|null); /** AttrValue shape */ shape?: (tensorflow_json.ITensorShape|null); /** AttrValue tensor */ tensor?: (tensorflow_json.ITensor|null); /** AttrValue placeholder */ placeholder?: (string|null); /** AttrValue func */ func?: (tensorflow_json.INameAttrList|null); } export namespace AttrValue { /** Properties of a ListValue. */ export interface IListValue { /** ListValue s */ s?: (string[]|null); /** ListValue i */ i?: ((number | string)[]|null); /** ListValue f */ f?: (number[]|null); /** ListValue b */ b?: (boolean[]|null); /** ListValue type */ type?: (tensorflow_json.DataType[]|null); /** ListValue shape */ shape?: (tensorflow_json.ITensorShape[]|null); /** ListValue tensor */ tensor?: (tensorflow_json.ITensor[]|null); /** ListValue func */ func?: (tensorflow_json.INameAttrList[]|null); } } /** Properties of a NameAttrList. */ export interface INameAttrList { /** NameAttrList name */ name?: (string|null); /** NameAttrList attr */ attr?: ({[k: string]: tensorflow_json.IAttrValue}|null); } /** Properties of a NodeDef. */ export interface INodeDef { /** NodeDef name */ name?: (string|null); /** NodeDef op */ op?: (string|null); /** NodeDef input */ input?: (string[]|null); /** NodeDef device */ device?: (string|null); /** NodeDef attr */ attr?: ({[k: string]: tensorflow_json.IAttrValue}|null); } /** Properties of a VersionDef. */ export interface IVersionDef { /** VersionDef producer */ producer?: (number|null); /** VersionDef minConsumer */ minConsumer?: (number|null); /** VersionDef badConsumers */ badConsumers?: (number[]|null); } /** Properties of a GraphDef. */ export interface IGraphDef { /** GraphDef node */ node?: (tensorflow_json.INodeDef[]|null); /** GraphDef versions */ versions?: (tensorflow_json.IVersionDef|null); /** GraphDef library */ library?: (tensorflow_json.IFunctionDefLibrary|null); } /** Properties of a CollectionDef. */ export interface ICollectionDef { /** CollectionDef nodeList */ nodeList?: (tensorflow_json.CollectionDef.INodeList|null); /** CollectionDef bytesList */ bytesList?: (tensorflow_json.CollectionDef.IBytesList|null); /** CollectionDef int64List */ int64List?: (tensorflow_json.CollectionDef.IInt64List|null); /** CollectionDef floatList */ floatList?: (tensorflow_json.CollectionDef.IFloatList|null); /** CollectionDef anyList */ anyList?: (tensorflow_json.CollectionDef.IAnyList|null); } export namespace CollectionDef { /** Properties of a NodeList. */ export interface INodeList { /** NodeList value */ value?: (string[]|null); } /** Properties of a BytesList. */ export interface IBytesList { /** BytesList value */ value?: (Uint8Array[]|null); } /** Properties of an Int64List. */ export interface IInt64List { /** Int64List value */ value?: ((number | string)[]|null); } /** Properties of a FloatList. */ export interface IFloatList { /** FloatList value */ value?: (number[]|null); } /** Properties of an AnyList. */ export interface IAnyList { /** AnyList value */ value?: (tensorflow_json.IAny[]|null); } } /** Properties of a SaverDef. */ export interface ISaverDef { /** SaverDef filenameTensorName */ filenameTensorName?: (string|null); /** SaverDef saveTensorName */ saveTensorName?: (string|null); /** SaverDef restoreOpName */ restoreOpName?: (string|null); /** SaverDef maxToKeep */ maxToKeep?: (number|null); /** SaverDef sharded */ sharded?: (boolean|null); /** SaverDef keepCheckpointEveryNHours */ keepCheckpointEveryNHours?: (number|null); /** SaverDef version */ version?: (tensorflow_json.SaverDef.CheckpointFormatVersion|null); } export namespace SaverDef { /** CheckpointFormatVersion enum. */ export enum CheckpointFormatVersion {LEGACY = 0, V1 = 1, V2 = 2} } /** Properties of a TensorInfo. */ export interface ITensorInfo { /** TensorInfo name */ name?: (string|null); /** TensorInfo cooSparse */ cooSparse?: (tensorflow_json.TensorInfo.ICooSparse|null); /** TensorInfo dtype */ dtype?: (tensorflow_json.DataType|null); /** TensorInfo tensorShape */ tensorShape?: (tensorflow_json.ITensorShape|null); } export namespace TensorInfo { /** Properties of a CooSparse. */ export interface ICooSparse { /** CooSparse valuesTensorName */ valuesTensorName?: (string|null); /** CooSparse indicesTensorName */ indicesTensorName?: (string|null); /** CooSparse denseShapeTensorName */ denseShapeTensorName?: (string|null); } } /** Properties of a SignatureDef. */ export interface ISignatureDef { /** SignatureDef inputs */ inputs?: ({[k: string]: tensorflow_json.ITensorInfo}|null); /** SignatureDef outputs */ outputs?: ({[k: string]: tensorflow_json.ITensorInfo}|null); /** SignatureDef methodName */ methodName?: (string|null); } /** Properties of an AssetFileDef. */ export interface IAssetFileDef { /** AssetFileDef tensorInfo */ tensorInfo?: (tensorflow_json.ITensorInfo|null); /** AssetFileDef filename */ filename?: (string|null); } /** Properties of an OpDef. */ export interface IOpDef { /** OpDef name */ name?: (string|null); /** OpDef inputArg */ inputArg?: (tensorflow_json.OpDef.IArgDef[]|null); /** OpDef outputArg */ outputArg?: (tensorflow_json.OpDef.IArgDef[]|null); /** OpDef attr */ attr?: (tensorflow_json.OpDef.IAttrDef[]|null); /** OpDef deprecation */ deprecation?: (tensorflow_json.OpDef.IOpDeprecation|null); /** OpDef summary */ summary?: (string|null); /** OpDef description */ description?: (string|null); /** OpDef isCommutative */ isCommutative?: (boolean|null); /** OpDef isAggregate */ isAggregate?: (boolean|null); /** OpDef isStateful */ isStateful?: (boolean|null); /** OpDef allowsUninitializedInput */ allowsUninitializedInput?: (boolean|null); } export namespace OpDef { /** Properties of an ArgDef. */ export interface IArgDef { /** ArgDef name */ name?: (string|null); /** ArgDef description */ description?: (string|null); /** ArgDef type */ type?: (tensorflow_json.DataType|null); /** ArgDef typeAttr */ typeAttr?: (string|null); /** ArgDef numberAttr */ numberAttr?: (string|null); /** ArgDef typeListAttr */ typeListAttr?: (string|null); /** ArgDef isRef */ isRef?: (boolean|null); } /** Properties of an AttrDef. */ export interface IAttrDef { /** AttrDef name */ name?: (string|null); /** AttrDef type */ type?: (string|null); /** AttrDef defaultValue */ defaultValue?: (tensorflow_json.IAttrValue|null); /** AttrDef description */ description?: (string|null); /** AttrDef hasMinimum */ hasMinimum?: (boolean|null); /** AttrDef minimum */ minimum?: (number|string|null); /** AttrDef allowedValues */ allowedValues?: (tensorflow_json.IAttrValue|null); } /** Properties of an OpDeprecation. */ export interface IOpDeprecation { /** OpDeprecation version */ version?: (number|null); /** OpDeprecation explanation */ explanation?: (string|null); } } /** Properties of an OpList. */ export interface IOpList { /** OpList op */ op?: (tensorflow_json.IOpDef[]|null); } /** Properties of a MetaGraphDef. */ export interface IMetaGraphDef { /** MetaGraphDef metaInfoDef */ metaInfoDef?: (tensorflow_json.MetaGraphDef.IMetaInfoDef|null); /** MetaGraphDef graphDef */ graphDef?: (tensorflow_json.IGraphDef|null); /** MetaGraphDef saverDef */ saverDef?: (tensorflow_json.ISaverDef|null); /** MetaGraphDef collectionDef */ collectionDef?: ({[k: string]: tensorflow_json.ICollectionDef}|null); /** MetaGraphDef signatureDef */ signatureDef?: ({[k: string]: tensorflow_json.ISignatureDef}|null); /** MetaGraphDef assetFileDef */ assetFileDef?: (tensorflow_json.IAssetFileDef[]|null); } export namespace MetaGraphDef { /** Properties of a MetaInfoDef. */ export interface IMetaInfoDef { /** MetaInfoDef metaGraphVersion */ metaGraphVersion?: (string|null); /** MetaInfoDef strippedOpList */ strippedOpList?: (tensorflow_json.IOpList|null); /** MetaInfoDef anyInfo */ anyInfo?: (tensorflow_json.IAny|null); /** MetaInfoDef tags */ tags?: (string[]|null); /** MetaInfoDef tensorflowVersion */ tensorflowVersion?: (string|null); /** MetaInfoDef tensorflowGitVersion */ tensorflowGitVersion?: (string|null); } } /** Properties of a SavedModel. */ export interface ISavedModel { /** SavedModel savedModelSchemaVersion */ savedModelSchemaVersion?: (number|string|null); /** SavedModel metaGraphs */ metaGraphs?: (tensorflow_json.IMetaGraphDef[]|null); } /** Properties of a FunctionDefLibrary. */ export interface IFunctionDefLibrary { /** FunctionDefLibrary function */ 'function'?: (tensorflow_json.IFunctionDef[]|null); /** FunctionDefLibrary gradient */ gradient?: (tensorflow_json.IGradientDef[]|null); } /** Properties of a FunctionDef. */ export interface IFunctionDef { /** FunctionDef signature */ signature?: (tensorflow_json.IOpDef|null); /** FunctionDef attr */ attr?: ({[k: string]: tensorflow_json.IAttrValue}|null); /** FunctionDef nodeDef */ nodeDef?: (tensorflow_json.INodeDef[]|null); /** FunctionDef ret */ ret?: ({[k: string]: string}|null); } /** Properties of a GradientDef. */ export interface IGradientDef { /** GradientDef functionName */ functionName?: (string|null); /** GradientDef gradientFunc */ gradientFunc?: (string|null); } }