vision
https://github.com/pytorch/vision
Python
Datasets, Transforms and Models specific to Computer Vision
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported17 Subscribers
Add a CodeTriage badge to vision
Help out
- Issues
- Harden LSUNClass key cache: store under dataset root and load with weights_only
- Guard roi_align/ps_roi_align bilinear sampling against non-finite boxes
- Bounds-check GIF palette indices in decode_gif
- Use https for dataset download URLs where the host supports TLS
- Default to weights_only=True when loading pretrained weights
- Fix/8877 elastic transform negative sigma
- Allow v2 Normalize to reuse tensor statistics
- Enable mypy type checking for torchvision.io
- Port roi_pool MPS kernel to the stable-ABI extension
- Port roi_align MPS kernel to the stable-ABI extension
- Docs
- Python not yet supported