tfjs-tiny-yolov2
Version:
Tiny YOLO v2 object detection with tensorflow.js.
9 lines (8 loc) • 386 B
TypeScript
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[];
};