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.
1 maintainer
Metadata
1 maintainer
Release files for descriptron-vision 2.6.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| descriptron_vision-2.6.0.tar.gz | 178.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| descriptron_vision-2.6.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 375.0 kB
Release files / descriptron_vision-2.6.0.tar.gz
| Download URL | descriptron_vision-2.6.0.tar.gz |
|---|---|
| Size | 178.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
35b4430005afc06653ad43ff438ae196522b90062fd0f321b6a4938fb7e237bd
|
|
BLAKE2b-256 checksum How to use checksums |
853f6674792a2f2ad7975dc8a5522c47f5c8107476703bef86ceea6f7e214cd4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.12.4
|
Release files / descriptron_vision-2.6.0-py3-none-any.whl
| Download URL | descriptron_vision-2.6.0-py3-none-any.whl |
|---|---|
| Size | 196.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d51f7c60f6f19209e45c6ea18ce29cfc38d500172d761b952d625c5fbac011cb
|
|
BLAKE2b-256 checksum How to use checksums |
99fcb68948b01c4d5d75695f37f51556b8c590d81836a9ba9376a9f3d38f5697
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.12.4
|
