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 supported15 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
- [DONOTMERGE] add CI tests for torch.compile'ing the transforms.v2 kernels
- [Tests] Added torch compile checks for functional ops tests
- add smoke test for prototype namespace on release versions
- Add masks to boundaries
- ViTDet object detection + segmentation implementation
- remove private imports from torch.testing
- ci: Limit macOS testing to one version of python
- YOLO
- pytorch conda channel is missing torchvision `0.2.2`
- Docs
- Python not yet supported