Supported NNs
Supervisely suppports most of the state of the art models for common computer vision tasks:
All neural networks architectures (listed below) support both training and inference inside the Supervisely Platform. Also we provide pretrained weights for each architecture that can be used directly for inference or for transfer learning to speed up the training process on your custom data.
We aslo provide source codes for all neural networks so developers can customize them. Read more in the SDK chapter.

Interactive segmentation

Architecture
Pretrained on
Framework
Paper
Smart Tool
Custom data
PyTorch
โ€‹Mediumโ€‹
โ€‹
Custom data + Supervisely Person
โ€‹
โ€‹Medium1, Medium2โ€‹

Semantic segmentation

Architecture
Pretrained on
Framework
Paper
Unet V2 (based on VGG)
ImageNet
PyTorch
โ€‹arxivโ€‹
PSPNet
Cityscapes
Tensorflow
โ€‹arxivโ€‹
โ€‹
ADE20K
โ€‹
โ€‹
ICNet
Cityscapes
Tensorflow
โ€‹arxivโ€‹
โ€‹
ADE20K
โ€‹
โ€‹
DeepLab V3 +
Cityscapes
Tensorflow
โ€‹arxivโ€‹
โ€‹
ADE20K
โ€‹
โ€‹

Instance segmentation

Architecture
Backbone
Pretrained on
Framework
Paper
Mask-RCNN
Inception V2
COCO
Tensorflow
โ€‹arxivโ€‹
โ€‹
ResNet-50 (atrous)
COCO
โ€‹
โ€‹
โ€‹
ResNet-101 (atrous)
COCO
โ€‹
โ€‹

Object detection

Architecture
Backbone
Pretrained on
Framework
Paper
YOLO V3
โ€‹
COCO
DarkNet (C++)
โ€‹arxivโ€‹
SSD
Inception V2
COCO
Tensorflow
โ€‹arxivโ€‹
โ€‹
MobileNet V1
COCO
โ€‹
โ€‹
โ€‹
MobileNet V2
COCO
โ€‹
โ€‹
โ€‹
MobileNet V2 (lite)
COCO
โ€‹
โ€‹
Faster-RCNN
ResNet-50
COCO
Tensorflow
โ€‹arxivโ€‹
โ€‹
ResNet-101
COCO
โ€‹
โ€‹
โ€‹
Inception V2
COCO
โ€‹
โ€‹
โ€‹
Inception ResNet V2 (atrous)
COCO
โ€‹
โ€‹
โ€‹
NasNet
COCO
โ€‹
โ€‹
โ€‹
NasNet (lowproposals)
COCO
โ€‹
โ€‹
RFCN
ResNet-101
COCO
Tensorflow
โ€‹arxivโ€‹

Classification

Architecture
Backbone
Pretrained on
Framework
Paper
AlexNet
โ€‹
ImageNet
PyTorch
โ€‹arxivโ€‹
VGG
*-11
ImageNet
PyTorch
โ€‹arxivโ€‹
โ€‹
*-13
ImageNet
โ€‹
โ€‹
โ€‹
*-16
ImageNet
โ€‹
โ€‹
โ€‹
*-19
ImageNet
โ€‹
โ€‹
โ€‹
*-11 with batchnorm
ImageNet
โ€‹
โ€‹
โ€‹
*-13 with batchnorm
ImageNet
โ€‹
โ€‹
โ€‹
*-16 with batchnorm
ImageNet
โ€‹
โ€‹
โ€‹
*-19 with batchnorm
ImageNet
โ€‹
โ€‹
ResNet
*-18
ImageNet
PyTorch
โ€‹arxivโ€‹
โ€‹
*-34
ImageNet
โ€‹
โ€‹
โ€‹
*-50
ImageNet
โ€‹
โ€‹
โ€‹
*-101
ImageNet
โ€‹
โ€‹
โ€‹
*-152
ImageNet
โ€‹
โ€‹
SqueezeNet
*-1.0
ImageNet
PyTorch
โ€‹arxivโ€‹
โ€‹
*-1.1
ImageNet
โ€‹
โ€‹
Densenet
*-121
ImageNet
PyTorch
โ€‹arxivโ€‹
โ€‹
*-169
ImageNet
โ€‹
โ€‹
โ€‹
*-201
ImageNet
โ€‹
โ€‹
โ€‹
*-161
ImageNet
โ€‹
โ€‹
Inception V3
โ€‹
ImageNet
PyTorch
โ€‹arxivโ€‹

Text detection

Architecture
Pretrained on
Framework
Paper
EAST
ICDAR 2015
Tensorflow
โ€‹arxivโ€‹
CTPN
ICDAR 2015
Tensorflow
โ€‹arxivโ€‹
PixelLink
ICDAR 2015
Tensorflow
โ€‹arxivโ€‹

OCR

Architecture
Pretrained on
Framework
Paper
CNN-LSTM-CTC
ICDAR 2015
Tensorflow
โ€‹arxivโ€‹
Last modified 1yr ago