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
- 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.
- [conda] Older Torchvision versions (<0.6.1) no longer available on PyTorch's conda channel
- Potential tricky bug in the implementation of stochastic depth
- Make read_video/write_video support Path from pathlib
- Docs
- Python not yet supported