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 supported14 Subscribers
Add a CodeTriage badge to vision
Help out
- Issues
- OOM Error with `roi_align` in PyTorch 2.1.1 but fine in PyTorch 2.0.1
- RuntimeError: cuda video backend is not available.
- Video decoding tests are never run on CI
- [DEBUG] run pytest without color
- [DEBUG] video extension not available in CI
- `to_pil_image` rounds down
- [DONOTMERGE] add CI tests for torch.compile'ing the transforms.v2 kernels
- Models input/output conventions (shape, preproc/postproc, etc.)
- Fix ONNX export of RAFT optical flow model
- MaxViT number of channel (input_channels) as a parameter
- Docs
- Python not yet supported