tfjs-tiny-yolov2
Version:
Tiny YOLO v2 object detection with tensorflow.js.
8 lines (7 loc) • 368 B
TypeScript
import { ParamMapping } from 'tfjs-image-recognition-base';
import { TinyYolov2Config } from './config';
import { TinyYolov2NetParams } from './types';
export declare function extractParams(weights: Float32Array, config: TinyYolov2Config, boxEncodingSize: number, filterSizes: number[]): {
params: TinyYolov2NetParams;
paramMappings: ParamMapping[];
};