2020-10-09 03:31:31 +02:00
|
|
|
import * as tf from '@tensorflow/tfjs';
|
2020-08-18 13:54:53 +02:00
|
|
|
|
|
|
|
import { NetInput } from './NetInput';
|
|
|
|
|
|
|
|
export type TMediaElement = HTMLImageElement | HTMLVideoElement | HTMLCanvasElement
|
|
|
|
|
|
|
|
export type TResolvedNetInput = TMediaElement | tf.Tensor3D | tf.Tensor4D
|
|
|
|
|
|
|
|
export type TNetInputArg = string | TResolvedNetInput
|
|
|
|
|
|
|
|
export type TNetInput = TNetInputArg | Array<TNetInputArg> | NetInput | tf.Tensor4D
|