The collection of pre-trained, state-of-the-art AI models.
ailia SDK is a cross-platform high speed inference SDK. The ailia SDK provides a consistent C++ API on Windows, Mac, Linux, iOS, Android, Jetson and Raspberry Pi. It supports Unity, Python and JNI for efficient AI implementation. The ailia SDK makes great use of the GPU via Vulkan and Metal to serve accelerated computing.
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
mars | MARS: Motion-Augmented RGB Stream for Action Recognition | Pytorch | 1.2.4 and later |
![]() |
st-gcn | ST-GCN | Pytorch | 1.2.5 and later |
![]() |
ax_action_recognition | Realtime-Action-Recognition | Pytorch | 1.2.7 and later |
![]() |
va-cnn | View Adaptive Neural Networks (VA) for Skeleton-based Human Action Recognition | Pytorch | 1.2.7 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
padim | PaDiM-Anomaly-Detection-Localization-master | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version |
---|---|---|---|
crnn_audio_classification | crnn-audio-classification | Pytorch | 1.2.5 and later |
deepspeech2 | deepspeech.pytorch | Pytorch | 1.2.2 and later |
pytorch-dc-tts | Efficiently Trainable Text-to-Speech System Based on Deep Convolutional Networks with Guided Attention | Pytorch | 1.2.6 and later |
unet_source_separation | source_separation | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia 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 and later |
![]() |
c-3-framework | Crowd Counting Code Framework(C^3-Framework) | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
clothing-detection | Clothing-Detection | Pytorch | 1.2.1 and later |
![]() |
mmfashion | MMFashion | Pytorch | 1.2.5 and later |
![]() |
fashionai-key-points-detection | A Pytorch Implementation of Cascaded Pyramid Network for FashionAI Key Points Detection | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
monodepth2 | Monocular depth estimation from a single image | Pytorch | 1.2.2 and later |
![]() |
midas | Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer |
Pytorch | 1.2.4 and later |
![]() |
fcrn-depthprediction | Deeper Depth Prediction with Fully Convolutional Residual Networks | TensorFlow | 1.2.6 and later |
![]() |
fast-depth | ICRA 2019 "FastDepth: Fast Monocular Depth Estimation on Embedded Systems" | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
yolov1-face | YOLO-Face-detection | Darknet | 1.1.0 and later |
![]() |
yolov3-face | Face detection using keras-yolov3 | Keras | 1.2.1 and later |
![]() |
blazeface | BlazeFace-PyTorch | Pytorch | 1.2.1 and later |
![]() |
face-mask-detection | Face detection using keras-yolov3 | Keras | 1.2.1 and later |
![]() |
dbface | DBFace : real-time, single-stage detector for face detection, with faster speed and higher accuracy |
Pytorch | 1.2.2 and later |
![]() |
retinaface | RetinaFace: Single-stage Dense Face Localisation in the Wild. | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
vggface2 | VGGFace2 Dataset for Face Recognition | Caffe | 1.1.0 and later |
![]() |
arcface | pytorch implement of arcface | Pytorch | 1.2.1 and later |
![]() |
insightface | InsightFace: 2D and 3D Face Analysis Project | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
face_classification | Real-time face detection and emotion/gender classification | Keras | 1.1.0 and later |
![]() |
facial_feature | kaggle-facial-keypoints | Pytorch | 1.2.0 and later |
![]() |
face_alignment | 2D and 3D Face alignment library build using pytorch | Pytorch | 1.2.1 and later |
![]() |
prnet | Joint 3D Face Reconstruction and Dense Alignment with Position Map Regression Network |
TensorFlow | 1.2.2 and later |
![]() |
gazeml | A deep learning framework based on Tensorflow for the training of high performance gaze estimation |
TensorFlow | 1.2.0 and later |
![]() |
facemesh | facemesh.pytorch | Pytorch | 1.2.2 and later |
![]() |
mediapipe_iris | irislandmarks.pytorch | Pytorch | 1.2.2 and later |
![]() |
hopenet | deep-head-pose | Pytorch | 1.2.2 and later |
![]() |
ax_gaze_estimation | ax Gaze Estimation | Pytorch | 1.2.2 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
flavr | FLAVR: Flow-Agnostic Video Representations for Fast Frame Interpolation | Pytorch | 1.2.7 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
pytorch-gan | Code repo for the Pytorch GAN Zoo project (used to train this model) | Pytorch | 1.2.4 and later |
![]() |
council-GAN | Council-GAN | Pytorch | 1.2.4 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
yolov3-hand | Hand detection branch of Face detection using keras-yolov3 | Keras | 1.2.1 and later |
![]() |
hand_detection_pytorch | hand-detection.PyTorch | Pytorch | 1.2.2 and later |
![]() |
blazepalm | MediaPipePyTorch | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
blazehand | MediaPipePyTorch | Pytorch | 1.2.5 and later |
![]() |
hand3d | ColorHandPose3D network | TensorFlow | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
illustration2vec | Illustration2Vec | Caffe | 1.2.2 and later |
![]() |
image_captioning_pytorch | Image Captioning pytorch | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
vgg16 | Very Deep Convolutional Networks for Large-Scale Image Recognition | Keras | 1.1.0 and later |
![]() |
googlenet | Going Deeper with Convolutions | Pytorch | 1.2.0 and later |
![]() |
resnet50 | Deep Residual Learning for Image Recognition | Chainer | 1.2.0 and later |
![]() |
inceptionv3 | Rethinking the Inception Architecture for Computer Vision | Pytorch | 1.2.0 and later |
![]() |
inceptionv4 | Keras Inception-V4 | Keras | 1.2.5 and later |
![]() |
mobilenetv2 | PyTorch Implemention of MobileNet V2 | Pytorch | 1.2.0 and later |
![]() |
mobilenetv3 | PyTorch Implemention of MobileNet V3 | Pytorch | 1.2.1 and later |
partialconv | Partial Convolution Layer for Padding and Image Inpainting | Pytorch | 1.2.0 and later | |
![]() |
efficientnet | A PyTorch implementation of EfficientNet | Pytorch | 1.2.3 and later |
![]() |
vit | Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale) | Pytorch | 1.2.7 and later |
![]() |
efficientnetv2 | EfficientNetV2 | Pytorch | 1.2.4 and later |
Model | Reference | Exported From | Supported Ailia Version |
---|---|---|---|
bert | pytorch-pretrained-bert | Pytorch | 1.2.2 and later |
bert_maskedlm | huggingface/transformers | Pytorch | 1.2.5 and later |
bert_ner | huggingface/transformers | Pytorch | 1.2.5 and later |
bert_question_answering | huggingface/transformers | Pytorch | 1.2.5 and later |
bert_sentiment_analysis | huggingface/transformers | Pytorch | 1.2.5 and later |
bert_zero_shot_classification | huggingface/transformers | Pytorch | 1.2.5 and later |
bert_tweets_sentiment | huggingface/transformers | Pytorch | 1.2.5 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
deepsort | Deep Sort with PyTorch | Pytorch | 1.2.3 and later |
![]() |
person_reid_baseline_pytorch | UTS-Person-reID-Practical | Pytorch | 1.2.6 and later |
![]() |
roneld | RONELD-Lane-Detection | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
pointnet_pytorch | PointNet.pytorch | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
rotnet | CNNs for predicting the rotation angle of an image to correct its orientation | Keras | 1.2.1 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
adain | Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization | Pytorch | 1.2.1 and later |
![]() |
psgan | PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer | Pytorch | 1.2.7 and later |
![]() |
beauty_gan | BeautyGAN | Pytorch | 1.2.7 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
srresnet | Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network | Pytorch | 1.2.0 and later |
![]() |
edsr | Enhanced Deep Residual Networks for Single Image Super-Resolution | Pytorch | 1.2.6 and later |
![]() |
han | Single Image Super-Resolution via a Holistic Attention Network | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
craft_pytorch | CRAFT: Character-Region Awareness For Text detection | Pytorch | 1.2.2 and later |
![]() |
pixel_link | Pixel-Link | TensorFlow | 1.2.6 and later |
![]() |
east | EAST: An Efficient and Accurate Scene Text Detector | TensorFlow | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version | |
---|---|---|---|---|
![]() |
etl | Japanese Character Classification | Keras | 1.1.0 and later |
![]() |
deep-text-recognition-benchmark | deep-text-recognition-benchmark | Pytorch | 1.2.6 and later |
![]() |
crnn.pytorch | Convolutional Recurrent Neural Network | Pytorch | 1.2.6 and later |
![]() |
paddleocr | PaddleOCR : Awesome multilingual OCR toolkits based on PaddlePaddle | Pytorch | 1.2.6 and later |
Model | Reference | Exported From | Supported Ailia Version |
---|---|---|---|
acculus-pose | Acculus, Inc. | Caffe | 1.2.3 and later |