UNPKG

tfjs-tiny-yolov2

Version:

Tiny YOLO v2 object detection with tensorflow.js.

9 lines (8 loc) 386 B
import * as tf from '@tensorflow/tfjs-core'; import { ParamMapping } from 'tfjs-image-recognition-base'; import { TinyYolov2Config } from './config'; import { TinyYolov2NetParams } from './types'; export declare function extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap, config: TinyYolov2Config): { params: TinyYolov2NetParams; paramMappings: ParamMapping[]; };