descriptron-vision

Mask and landmark prediction for Descriptron: torchvision detectors, SAM2-PAL propagation and DINOv3 landmark transfer.

pip install descriptron-vision
descriptron-train --task masks --coco-json annotations.json --img-dir images/ \
                  --output-dir out/ --total-iters 20000
descriptron-predict --checkpoint out/model_final_*.pth --coco-json annotations.json \
                    --img-dir images/ --annotations_out predictions.json

No compiler required. torch and torchvision are ordinary wheels on Linux, macOS and Windows. For CUDA, install torch from PyTorch's own index first:

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install descriptron-vision

On a Mac the detectors can use the Apple GPU through PyTorch's MPS device, which is why this backend exists: Detectron2 has no macOS wheels and no Apple Silicon support.

Two optional dependencies that cannot be declared

PyPI rejects a git dependency in package metadata, so neither SAM2 nor Detectron2 is listed. Both are imported lazily, and the error names the command to install them. The Docker image carries both ready to run.

Choosing a backbone

--arch v1 (default) is maskrcnn_resnet50_fpn; --arch v2 is maskrcnn_resnet50_fpn_v2. v2 scores higher on COCO, but on a few hundred training images v1 won on 3 of 3 folds of a grouped species hold-out — its lighter two-FC box head converges further in the same budget. Measured on the data, not assumed from the model zoo.

Metadata

Release files for descriptron-vision 2.2.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for descriptron-vision 2.2.0
File	Size	Uploaded
descriptron_vision-2.2.0.tar.gz	178.8 kB	Sep 30, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for descriptron-vision 2.2.0
File	Interpreter	ABI	Platform	Reset
descriptron_vision-2.2.0-py3-none-any.whl 196.2 kB Sep 30, 2026	Python 3	none	any	Details

Total release size: 375.0 kB

Release files / descriptron_vision-2.2.0.tar.gz

Download URL	descriptron_vision-2.2.0.tar.gz
Size	178.8 kB
Tags	Source
SHA-256 checksum How to use checksums	`0d0a86483fbf360b1a066bd0db4c40af78fe77a2e6c3a05d8e67ba7cf66188cf`
BLAKE2b-256 checksum How to use checksums	`ebf869012d1b1fef8c4c4238abde72babe8b639100e3dd73d0e42a267b5e5fa6`
Upload date	Sep 30, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.12.4`

Release files / descriptron_vision-2.2.0-py3-none-any.whl

Download URL	descriptron_vision-2.2.0-py3-none-any.whl
Size	196.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`13258b666ccd4c3fc8ce83b93d78627f41d2022baa7b7f0923b529c7e8a17d33`
BLAKE2b-256 checksum How to use checksums	`5128aee9a917dcebd39b7509ac0a4be78266179e9397fdb9a91f54f8800a8a9a`
Upload date	Sep 30, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.12.4`