extractParamsFromWeigthMap.d.ts 366 B

12345678
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { ParamMapping } from '../common/types';
  3. import { TinyYolov2Config } from './config';
  4. import { TinyYolov2NetParams } from './types';
  5. export declare function extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap, config: TinyYolov2Config): {
  6. params: TinyYolov2NetParams;
  7. paramMappings: ParamMapping[];
  8. };