Skip to content
Pretrained models for ailia SDK
Python
Branch: master
Clone or download
Latest commit 78d9c63 Jan 23, 2020
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
deeplabv3
dewarpnet
face_alignment
face_classification
facial_feature
gazeml
googlenet
hair_segmentation
hrnet_segmentation
inceptionv3
lightweight-human-pose-estimation
mobilenet_ssd
mobilenetv2
mobilenetv3
noise2noise
openpose
partialconv
resnet50
rotnet
srresnet
vgg16
vggface2
yolov1-face
yolov1-tiny
yolov2
yolov3-face
yolov3-tiny
yolov3
.gitattributes
README.md

README.md

ailia-models

The collection of pre-trained, state-of-the-art models.

Image classification

Name Detail Exported From Supported Version
vgg16 Very Deep Convolutional Networks for Large-Scale Image Recognition Keras 1.1.0
googlenet Going Deeper with Convolutions Pytorch 1.2.0
resnet50 Deep Residual Learning for Image Recognition Chainer 1.2.0
inceptionv3 Rethinking the Inception Architecture for Computer Vision Pytorch 1.2.0
mobilenetv2 PyTorch Implemention of MobileNet V2 Pytorch 1.2.0
mobilenetv3 PyTorch Implemention of MobileNet V3 Pytorch 1.2.1 beta
partialconv Partial Convolution Layer for Padding and Image Inpainting Pytorch 1.2.0

Image segmentation

Name Detail Exported From Supported Version
deeplabv3 Xception65 for backbone network of DeepLab v3+ Chainer 1.2.0
hrnet_segmentation High-resolution networks (HRNets) for Semantic Segmentation Pytorch 1.2.1 beta
hair_segmentation hair segmentation in mobile device Keras 1.2.1 beta

Image manipulation

Name Detail Exported From Supported Version
srresnet Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network Pytorch 1.2.0
noise2noise Learning Image Restoration without Clean Data Pytorch 1.2.0
dewarpnet DewarpNet: Single-Image Document Unwarping With Stacked 3D and 2D Regression Networks Pytorch 1.2.1 beta

Object detection

Name Detail Exported From Supported Version
yolov1-tiny YOLO: Real-Time Object Detection Darknet 1.1.0
yolov1-face YOLO-Face-detection Darknet 1.1.0
yolov2 YOLO: Real-Time Object Detection Pytorch 1.2.0
yolov3 YOLO: Real-Time Object Detection ONNX Runtime 1.2.1 beta
yolov3-tiny YOLO: Real-Time Object Detection Keras 1.2.1 beta
yolov3-face Face detection using keras-yolov3 Keras 1.2.1 beta
mobilenet_ssd MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch Pytorch 1.2.1 beta

Pose estimation

Name Detail Exported From Supported Version
openpose Code repo for realtime multi-person pose estimation in CVPR'17 (Oral) Caffe 1.2.1 beta
lightweight-human-pose-estimation Fast and accurate human pose estimation in PyTorch. Contains implementation of "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose" paper. Pytorch 1.2.0

Gaze estimation

Name Detail Exported From Supported Version
gazeml A deep learning framework based on Tensorflow for the training of high performance gaze estimation TensorFlow 1.2.0

Face recognization

Name Detail Exported From Supported Version
face_classification Real-time face detection and emotion/gender classification Keras 1.1.0
vggface2 VGGFace2 Dataset for Face Recognition Caffe 1.1.0
facial_feature kaggle-facial-keypoints Pytorch 1.2.0
face_alignment 2D and 3D Face alignment library build using pytorch Pytorch 1.2.1 beta

Rotation estimation

Name Detail Exported From Supported Version
rotnet CNNs for predicting the rotation angle of an image to correct its orientation Keras 1.2.1 beta
You can’t perform that action at this time.