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
- 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
- libtorch C++, fasterrcnn_resnet50_fpn module.forward() Assert
- Multi-class focal loss
- Add -O3 compilation flags
- [RFC] Simplify release note creation
- Custom transform support
- does deformable conv support export to onnx?
- Error(s) in loading state_dict in ..., unexpected keys
- Docs
- Python not yet supported