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
- torchvision.io.read_image should take file path or image encoded as bytes
- Caltech-101 doesn't load bboxes but it should.
- Generalized IOU loss doesn't handle 0-length and -width boxes correctly, expected 0.0 but got 1.0
- GPU decoder with 'cuda' backend not working as expected (shape / color / seek)
- Possibility to have saved models including their optimizer and scheduler state_dicts
- roi_align and roi_pool: Allow separate spatial scales for inputs of different height and width
- Update transforms.py
- unittests on Linux GPU often run out of memory
- Segmentation fault in JIT tests on Windows CPU
- Add a data augmentation method PatchMix for Contrastive Learning and ViT.
- Docs
- Python not yet supported