Options
All
  • Public
  • Public/Protected
  • All
Menu

Hierarchy

  • TinyYolov2Base
    • TinyYolov2

Index

Constructors

  • new TinyYolov2(withSeparableConvs?: boolean): TinyYolov2

Properties

_name: any
DEFAULT_FILTER_SIZES: number[] = ...

Accessors

  • get boxEncodingSize(): number
  • get isLoaded(): boolean
  • get paramMappings(): ParamMapping[]
  • Returns ParamMapping[]

  • get params(): undefined | TNetParams
  • Returns undefined | TNetParams

  • get withClassScores(): boolean
  • get withSeparableConvs(): boolean

Methods

  • dispose(throwOnRedispose?: boolean): void
  • Parameters

    • throwOnRedispose: boolean = true

    Returns void

  • extractWeights(weights: Float32Array): void
  • Parameters

    • weights: Float32Array

    Returns void

  • forward(input: any, inputSize: number): Promise<Tensor4D>
  • forwardInput(input: NetInput, inputSize: number): Tensor4D
  • freeze(): void
  • getFrozenParams(): { path: string; tensor: Tensor }[]
  • Returns { path: string; tensor: Tensor }[]

  • getParamFromPath(paramPath: string): Tensor
  • Parameters

    • paramPath: string

    Returns Tensor

  • getParamList(): { path: string; tensor: Tensor }[]
  • Returns { path: string; tensor: Tensor }[]

  • getTrainableParams(): { path: string; tensor: Tensor }[]
  • Returns { path: string; tensor: Tensor }[]

  • load(weightsOrUrl: undefined | string | Float32Array): Promise<void>
  • Parameters

    • weightsOrUrl: undefined | string | Float32Array

    Returns Promise<void>

  • loadFromDisk(filePath: undefined | string): Promise<void>
  • Parameters

    • filePath: undefined | string

    Returns Promise<void>

  • loadFromUri(uri: undefined | string): Promise<void>
  • Parameters

    • uri: undefined | string

    Returns Promise<void>

  • loadFromWeightMap(weightMap: NamedTensorMap): void
  • Parameters

    • weightMap: NamedTensorMap

    Returns void

  • reassignParamFromPath(paramPath: string, tensor: Tensor): void
  • Parameters

    • paramPath: string
    • tensor: Tensor

    Returns void

  • serializeParams(): Float32Array
  • Returns Float32Array

  • variable(): void