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
- [RFC]A potential solution to address the issue #9394 by by enlarging grid to (H+1, W+1) - seeking feedback
- Add torchvision logo
- Make FasterRCNN_mobilenet exportable via torch.export with dynamic shapes
- Investigate merging box conversion utils from ops and transforms
- Memory leak / unbounded RSS growth in torchvision.io.image.decode_jpeg() on malformed JPEG (CPU) → potential DoS
- Type hints of torchvision are not constistent with implementation
- CPU segfault in `torchvision.ops.roi_align` when ROIs contain NaN due to missing finite-value checks in `roi_align_forward_kernel.cpp`
- TorchVision Detection Models fail to export to onnx with dynamo=True
- The URL for the SUN397 dataset isn't working
- Setting the tensors and ndarrays of the 2, 3 and 4 channels of `int64` and `int32` to `ToPILImage()` with `None` for `mode` argument gets errors
- Docs
- Python not yet supported