tfjs-tiny-yolov2
Version:
Tiny YOLO v2 object detection with tensorflow.js.
5 lines (4 loc) • 521 B
TypeScript
import { ExtractWeightsFunction, ParamMapping } from 'tfjs-image-recognition-base';
import { SeparableConvParams } from './types';
export declare function extractSeparableConvParamsFactory(extractWeights: ExtractWeightsFunction, paramMappings: ParamMapping[]): (channelsIn: number, channelsOut: number, mappedPrefix: string) => SeparableConvParams;
export declare function loadSeparableConvParamsFactory(extractWeightEntry: <T>(originalPath: string, paramRank: number) => T): (prefix: string) => SeparableConvParams;