search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.2k stars 6.95k forks source link

Pretrained Group Normalization Models #631

Closed mingruimingrui closed 4 years ago

mingruimingrui commented 6 years ago

Hi, would there be any plans to be making pretrained group normalization models?

fmassa commented 6 years ago

Yes, thats a good feature request. We will be adding more models in torchvision soon. I'll open a master issue to keep track of them

qiulesun commented 5 years ago

Hi, the mainstream segmentation models, like EncNet (CVPR2018), PSPNet (CVPR2017) and Deeplab series, will be added or provide the examples?

fmassa commented 4 years ago

@qiulesun we have added pre-trained DeepLabV3 in torchvision with the 0.3 release, together with object detection, instance segmentation and person keypoint detection models