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
- Avoid overflow in box center conversion
- Fix pil_to_tensor for 16-bit images
- Widen float32 tolerance for TestRotatedBoxIou::test_iou on arm64 Linux
- Preserve channels-last memory format in to_image
- Fix two OOM error-path defects in the CPU PNG encoder
- 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
- Docs
- Python not yet supported