[Master Issue] Add more models to torchvision
See original GitHub issueThis is a master issue to track requests for adding new pre-trained models to torchvision.
Here is the (potentially incomplete) list I compiled:
- ResNext https://github.com/pytorch/vision/issues/154 https://github.com/pytorch/vision/issues/543 https://github.com/pytorch/vision/issues/154
- ResNet / ResNext with Group Norm https://github.com/pytorch/vision/issues/631
- mobilenet https://github.com/pytorch/vision/issues/625
- Inception family https://github.com/pytorch/vision/issues/490 https://github.com/pytorch/vision/issues/537
- ~NasNet https://github.com/pytorch/vision/issues/321~ Implemented as MNasNet
- SENet https://github.com/pytorch/vision/issues/260
- ShuffleNet
@Cadene has already implemented a number of those models in his fantastic https://github.com/Cadene/pretrained-models.pytorch . I’ll start from there and try to get models trained using pytorch/examples/imagenet, so that the models are reproducible.
Requirements
- python implementation to live in
vision/models - pre-trained weights using the same mean / std normalization as in the imagenet example
- the script used to train the models, or the command-line arguments used if the script was exactly the one from
examples/imagenet.
Issue Analytics
- State:
- Created 5 years ago
- Reactions:8
- Comments:46 (27 by maintainers)
Top Results From Across the Web
torchvision.models - PyTorch
The models subpackage contains definitions of models for addressing different tasks, including: image classification, pixelwise semantic segmentation, object ...
Read more >Getting Started with PyTorch Image Models (timm)
The purpose of this guide is to explore timm from a practitioner's point of view, focusing on how to use some of the...
Read more >Getting started with PyTorch - IBM
pytorch - Installs the GPU-enabled variants of PyTorch, torchvision, ... One or more elements of a deep learning model can lead to GPU...
Read more >timm - PyPI
(Unofficial) PyTorch Image Models. ... Add new RelPosMlp MaxViT weight that leverages this: ... Two more model weights added in the TPU trained...
Read more >torch.onnx — PyTorch master documentation
Using dictionaries to handle Named Arguments as model inputs. Indexing. Getter. Setter. TorchVision support. Limitations. Supported operators. Adding ...
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found

Sooo let me evaluate it after christmas to see which dataset would be better
@fmassa VOC and Cityscapes dataset is large, there is a smaller dataset CamVid consisting 701 labeled images, and if you use U-Net for better performance, I think using the original medical dataset is better. Here is a good project named pytorch-semseg for semantic segmentation reimplementing U-Net.