Available architectures

Here is the complete list of all the neural network architectures available in Studio. When available, links to the research papers are provided.

Classification architectures

Architecture

Trained on

Research paper

AlexNet

ImageNet

NIPS 2018

GoogleNet

ImageNet

arXiv

GoogleNet-Grayscale

ImageNet - grayscale

Inception-Resnet v2

ImageNet

ResNet-18

ImageNet

arXiv

ResNet-50

ImageNet

Inception v1

ImageNet

Inception v2

ImageNet

Inception v3

ImageNet

Inception v4

ImageNet

Detection architectures

Architecture

Model

Trained On

Research paper

Yolo

v2

VOC 2007

arXiv

v3

ImageNet 2012

arXiv

Faster-RCNN

VGG16

COCO 2018

arXiv

ZF

COCO 2018

arXiv

Resnet-50

COCO 2018

Resnet-101

COCO 2018

Inception Resnet v2

COCO 2018

RFCN

Resnet-101

COCO 2018

arXiv

SSD

Inception v2

COCO 2018

arXiv

MobileNet v1

COCO 2018

MobileNet v2

COCO 2018

SSDLite

MobileNet v2

COCO 2018

arXiv