sketchkit.cleanup package¶

Subpackages¶

class sketchkit.cleanup.cleaner.Cleaner(method: str = 'MasterSketch', device: str = 'cuda')[source]¶

Bases: object

A class for rough sketch cleanup.

This class provides functionality to convert a raster rough sketch to a clean sketch.

method¶

The vectorization method that is used.

device¶

The device that is used.

run(input: ndarray) → ndarray[source]¶

Simplify the input raster image.

Parameters:: input (numpy.ndarray) – input raster image in shape (H, W), with values in [0, 255].
Returns:: an image in H*W in [0, 255].
Return type:: sketch (numpy.ndarray)

class sketchkit.cleanup.Cleaner(method: str = 'MasterSketch', device: str = 'cuda')[source]¶