Class ArbitraryStyleTransferNetwork

Main ArbitraryStyleTransferNetwork class

Hierarchy

ArbitraryStyleTransferNetwork

Index

Constructors

constructor

Methods

Constructors

constructor

new ArbitraryStyleTransferNetwork(styleCheckpointURL?: string, transformCheckpointURL?: string): ArbitraryStyleTransferNetwork

- Defined in model.ts:41
ArbitraryStyleTransferNetwork constructor.

Parameters
- Default value styleCheckpointURL: string = DEFAULT_STYLE_CHECKPOINT
  
  Path to style model checkpoint directory.
- Default value transformCheckpointURL: string = DEFAULT_TRANSFORM_CHECKPOINT
  
  Path to transformation model checkpoint directory.
Returns ArbitraryStyleTransferNetwork

Methods

dispose

dispose(): void

- Defined in model.ts:83
Returns void

initialize

initialize(): Promise<void>

- Defined in model.ts:67
Loads models from the checkpoints.

Returns Promise<void>

isInitialized

isInitialized(): boolean

- Defined in model.ts:60
Returns true if model is initialized.

Returns boolean

stylize

stylize(content: ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement, style: ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement, strength?: number): Promise<ImageData>

- Defined in model.ts:140
This function stylizes the content image given the style image. It returns an ImageData instance containing the stylized image.

TODO(vdumoulin): Add option to resize style and content images. TODO(adarob): Add option to use model with depthwise separable convs.

Parameters
- content: ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement
  
  Content image to stylize
- style: ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement
  
  Style image to use
- Optional strength: number
  
  If provided, controls the stylization strength. Should be between 0.0 and 1.0.
Returns Promise<ImageData>

Generated using TypeDoc