Skip to content
Pretrained models for ailia SDK
Python
Branch: master
Clone or download

Files

Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
crowdcount-cascaded-mtl Added netron links Feb 11, 2020
deeplabv3 Added netron links Feb 11, 2020
dewarpnet Added netron links Feb 11, 2020
etl Added netron links Feb 11, 2020
face_alignment Added netron links Feb 11, 2020
face_classification
facial_feature
gazeml Added netron links Feb 11, 2020
googlenet Added netron link Feb 11, 2020
hair_segmentation Added netron links Feb 11, 2020
hrnet_segmentation Added netron links Feb 11, 2020
inceptionv3 Added netron link Feb 11, 2020
lightweight-human-pose-estimation
mobilenet_ssd Added netron links Feb 11, 2020
mobilenetv2
mobilenetv3 Added netron link Feb 11, 2020
noise2noise Added sample code of adding some noise to input image Feb 24, 2020
openpose Update output image of OpenPose Feb 16, 2020
partialconv Added netron links Feb 11, 2020
resnet50 Added netron link Feb 11, 2020
rotnet Added netron links Feb 11, 2020
srresnet Added netron links Feb 11, 2020
vgg16 Added netron link Feb 11, 2020
vggface2 Added netron links Feb 11, 2020
yolov1-face
yolov1-tiny Added netron link Feb 11, 2020
yolov2 Added netron link Feb 11, 2020
yolov3-face Added netron link Feb 11, 2020
yolov3-tiny Added video mode Feb 16, 2020
yolov3 Added netron link Feb 11, 2020
.gitattributes Add DMS models Sep 10, 2019
README.md Use pose estimator API for lw human pose Feb 16, 2020

README.md

ailia-models

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

About ailia SDK

ailia SDK is a cross-platform high speed inference SDK. The ailia SDK provides a consistent C++ API on Windows, Mac, Linux, iOS and Android. It supports Unity, Python and JNI for efficient AI implementation. The ailia SDK makes great use of the GPU to serve accelerated computing.

https://ailia.jp/en/

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.1 beta

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

Crowd counting

Name Detail Exported From Supported Version
crowdcount-cascaded-mtl CNN-based Cascaded Multi-task Learning of High-level Prior and Density Estimation for Crowd Counting (Single Image Crowd Counting) Pytorch 1.2.1 beta

Text recognization

Name Detail Exported From Supported Version
etl Japanese Character Recognization Keras 1.1.0
You can’t perform that action at this time.