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
- VideoClips: audio clips do not correspond to video clips
- More ResNet backbones for Faster R-CNN and Mask R-CNN
- [feature proposal] U-Nets with pretrained torchvision backbones
- [proposal] Use self.flatten instead of torch.flatten and when becomes possible derive ResNet from nn.Sequential (scripting+quantization is blocker), would simplify model surgery in the most frequent cases
- CapsNet Model
- libtorch C++, fasterrcnn_resnet50_fpn module.forward() Assert
- Multi-class focal loss
- VideoClips.get_clip to provide to users some clipId and a way to get meta information about clip pts/time in video
- 'download=True' condition for more than 1 SBD dataset stops code cause of shutil
- Error Compiling from Source with GPU on MacOS
- Docs
- Python not yet supported