sketchkit.image2sketch.HEDModel¶
- class sketchkit.image2sketch.HEDModel(*, checkpoint_dir: str | PathLike | None = None, auto_download: bool = True, postprocess: bool = True, postprocess_threshold: float = 0.09803921568627451, postprocess_small_edge: int = 5)[source]¶
High level wrapper around the original HED Caffe model.
- __init__(*, checkpoint_dir: str | PathLike | None = None, auto_download: bool = True, postprocess: bool = True, postprocess_threshold: float = 0.09803921568627451, postprocess_small_edge: int = 5) None[source]¶
Methods
__init__(*[, checkpoint_dir, auto_download, ...])Ensure the pretrained weights are available locally.
generate(image, *[, size])Generate edge maps for the provided image(s).
Attributes