sketchkit.image2sketch.anilines package

Subpackages

Submodules

sketchkit.image2sketch.anilines.generate module

class sketchkit.image2sketch.anilines.generate.AniLinesModel(*, checkpoint_dir: str | PathLike | None = None, mode: str = 'basic', device: str | device | None = None, binarize: float = -1.0, fp16: bool = True)[source]

Bases: object

High level wrapper around the AniLines LineExtractor.

DEFAULT_CHECKPOINT_DIR = PosixPath('/home/euruson/.cache/sketchkit/image2sketch/anilines')
_infer_single(image: Image, size: int | Tuple[int, int] | None = None) → Image[source]
_initialise_net() → Module[source]
_model: Module | None
_resolve_model_path() → Path[source]
_to_pil_image(image: Image | ndarray | str | PathLike) → Image[source]
generate(image: Image | ndarray | str | PathLike | Sequence[Image | ndarray | str | PathLike], *, size: int | Tuple[int, int] | None = None) → Image | List[Image][source]
sketchkit.image2sketch.anilines.generate.increase_sharpness(img: ndarray, factor: float = 6.0) → ndarray[source]

Module contents

class sketchkit.image2sketch.anilines.AniLinesModel(*, checkpoint_dir: str | PathLike | None = None, mode: str = 'basic', device: str | device | None = None, binarize: float = -1.0, fp16: bool = True)[source]

Bases: object

High level wrapper around the AniLines LineExtractor.

DEFAULT_CHECKPOINT_DIR = PosixPath('/home/euruson/.cache/sketchkit/image2sketch/anilines')
_infer_single(image: Image, size: int | Tuple[int, int] | None = None) → Image[source]
_initialise_net() → Module[source]
_model: Module | None
_resolve_model_path() → Path[source]
_to_pil_image(image: Image | ndarray | str | PathLike) → Image[source]
generate(image: Image | ndarray | str | PathLike | Sequence[Image | ndarray | str | PathLike], *, size: int | Tuple[int, int] | None = None) → Image | List[Image][source]