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
- Fix DefaultBoxGenerator CUDA device mismatch in _grid_default_boxes (#9414)
- Fix type hint inconsistencies in transforms/functional.py
- Fix docs build failure on Python 3.13+ by upgrading Sphinx
- viable/strict updating silently broken
- `TestRotatedBoxIou::test_iou[xyxyxyxy-float32]` fails on arm64 Linux due to float32 precision in format round-trip
- fix(transforms): handle float16/bfloat16 in convert_image_dtype
- Fix not exporting encode_png on windows
- Rotated bounding box NMS implementation for GPU
- Fix docstring style to match torchvision guidelines
- Problem with mask
- Docs
- Python not yet supported