Standard torchvision models.

[NiklasGustafsson]

If anyone is looking for ideas on what to help with, adding more of the standard models from 'torchvision.models' would be excellent.

I started on some of it already but porting the code from Python take a bit of time, so if anyone is interested in helping out, that would be excellent.

Niklas

[GeorgeS2019]

PyTorch Image and Video => Torchvision

TORCHVISION.MODELS

Classification

• AlexNet

• VGG

• ResNet

• SqueezeNet

• DenseNet

• Inception v3

• GoogLeNet

• ShuffleNet v2

• MobileNetV2

• MobileNetV3

• ResNeXt

• Wide ResNet

• MNASNet

• EfficientNet

• RegNet

Semantic Segmentation

Object Detection, Instance Segmentation and Person Keypoint Detection

Video classification

[GeorgeS2019]

@NiklasGustafsson What are Standard torchvision models?

[NiklasGustafsson]

https://pytorch.org/vision/stable/models.html

[GeorgeS2019]

@NiklasGustafsson I think we are on the same page