canny-chenn 1.0.1

A simple Canny edge detector for pytorch

Canny Edges

 

A simple Torch.nn.Module to return an image mask representing edges found by the Canny Edge-Finding algorithm.

Supports:

• PyTorch Tensor images formatted in image batches [B, C, h, w], with float32 data, and values spanning [0,1]
• any number of channels C
• TorchScript jit script, as well as the newer, traced torch-export, on the Canny nn.Module class

---

Usage

I use mise to handle project tools. Install it if you don't have it

git clone https://github.com/chrishenn/canny.git
cd canny
mise i

# list available just recipes
just --list

Available recipes:
    interactive # Detect and display canny edges on a set of test images. Use ctrl+c to kill
    lint        # run project formatters/linters [alias: l]
    test        # run basic tests
    unsafe      # run ruff with unsafe-fixes

---

Dev

# build distrobution pkgs locally
uv build --package canny_chenn

# push tag to trigger pypi to build pkgs and publish to pypi
git tag -a v1.0.0 -m v1.0.0 -f && git push --tags -f