human/types/object/centernet.d.ts

9 lines
315 B
TypeScript

/**
* CenterNet object detection module
*/
import { Item } from '../result';
import { GraphModel, Tensor } from '../tfjs/types';
import { Config } from '../config';
export declare function load(config: Config): Promise<GraphModel>;
export declare function predict(input: Tensor, config: Config): Promise<Item[]>;