Skip to content

axinc-ai/ailia-models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About ailia SDK

ailia SDK is a self-contained, cross-platform, high-speed inference SDK for AI. The ailia SDK provides a consistent C++ API across Windows, Mac, Linux, iOS, Android, Jetson, and Raspberry Pi platforms. It also supports Unity (C#), Python, Rust, Flutter(Dart) and JNI for efficient AI implementation. The ailia SDK makes extensive use of the GPU through Vulkan and Metal to enable accelerated computing.

How to use

Try now on Google Colaboratory

If you would like to try on your computer:

ailia MODELS tutorial

ailia MODELS tutorial 日本語版

Supported models

360 models as of November 19th, 2024

Latest update

  • 2024.11.19 Add qwen2_vl, live_portrait (ailia SDK 1.5)
  • 2024.10.09 Add whisper-v3-turbo
  • 2024.10.02 Add florence2
  • 2024.09.15 Add bert-vits2, pytorch_wavenet
  • 2024.09.12 Add gpt-sovits-v2
  • 2024.09.10 Add segment-anything-2 (video mode)
  • More information in our Wiki

Action recognition

Model Reference Exported From Supported Ailia Version Blog
mars MARS: Motion-Augmented RGB Stream for Action Recognition Pytorch 1.2.4 and later EN JP
st-gcn ST-GCN Pytorch 1.2.5 and later EN JP
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
driver-action-recognition-adas driver-action-recognition-adas-0002 OpenVINO 1.2.5 and later
action_clip ActionCLIP Pytorch 1.2.7 and later

Anomaly detection

Model Reference Exported From Supported Ailia Version Date Blog
mahalanobisad MahalanobisAD-pytorch Pytorch 1.2.9 and later May 2020
spade-pytorch Sub-Image Anomaly Detection with Deep Pyramid Correspondences Pytorch 1.2.6 and later May 2020
padim PaDiM-Anomaly-Detection-Localization-master Pytorch 1.2.6 and later Nov 2020 EN JP
patchcore PatchCore_anomaly_detection Pytorch 1.2.6 and later Jun 2021

Audio processing

Audio classification

Model Reference Exported From Supported Ailia Version Blog
crnn_audio_classification crnn-audio-classification Pytorch 1.2.5 and later EN JP
transformer-cnn-emotion-recognition Combining Spatial and Temporal Feature Representions of Speech Emotion by Parallelizing CNNs and Transformer-Encoders Pytorch 1.2.5 and later
audioset_tagging_cnn PANNs: Large-Scale Pretrained Audio Neural Networks for Audio Pattern Recognition Pytorch 1.2.9 and later
clap CLAP Pytorch 1.2.6 and later
microsoft clap CLAP Pytorch 1.2.11 and later

Music enhancement

Model Reference Exported From Supported Ailia Version Blog
hifigan HiFi-GAN Pytorch 1.2.9 and later
deep music enhancer On Filter Generalization for Music Bandwidth Extension Using Deep Neural Networks Pytorch 1.2.6 and later

Music generation

Model Reference Exported From Supported Ailia Version Blog
pytorch_wavenet pytorch_wavenet Pytorch 1.2.14 and later

Noise reduction

Model Reference Exported From Supported Ailia Version Blog
unet_source_separation source_separation Pytorch 1.2.6 and later EN JP
voicefilter VoiceFilter Pytorch 1.2.7 and later EN JP
rnnoise rnnoise Keras 1.2.15 and later
dtln Dual-signal Transformation LSTM Network Tensorflow 1.3.0 and later

Phoneme alignment

Model Reference Exported From Supported Ailia Version Blog
narabas narabas: Japanese phoneme forced alignment tool Pytorch 1.2.11 and later

Pitch detection

Model Reference Exported From Supported Ailia Version Blog
crepe torchcrepe Pytorch 1.2.10 and later JP

Speaker diarization

Model Reference Exported From Supported Ailia Version Blog
auto_speech AutoSpeech: Neural Architecture Search for Speaker Recognition Pytorch 1.2.5 and later EN JP
wespeaker WeSpeaker Onnxruntime 1.2.9 and later
pyannote-audio Pyannote-audio Pytorch 1.2.15 and later JP

Speech to text

Model Reference Exported From Supported Ailia Version Date Blog
deepspeech2 deepspeech.pytorch Pytorch 1.2.2 and later Oct 2017 EN JP
whisper Whisper Pytorch 1.2.10 and later Dec 2022 JP
reazon_speech ReazonSpeech Pytorch 1.4.0 and later Jan 2023
distil-whisper Hugging Face - Distil-Whisper Pytorch 1.2.16 and later Nov 2023
reazon_speech2 ReazonSpeech2 Pytorch 1.4.0 and later Feb 2024
kotoba-whisper kotoba-whisper Pytorch 1.2.16 and later Apr 2024

Text to speech

Model Reference Exported From Supported Ailia Version Date Blog
pytorch-dc-tts Efficiently Trainable Text-to-Speech System Based on Deep Convolutional Networks with Guided Attention Pytorch 1.2.6 and later Oct 2017 EN JP
tacotron2 Tacotron2 Pytorch 1.2.15 and later Feb 2018 JP
vall-e-x VALL-E-X Pytorch 1.2.15 and later Mar 2023 JP
Bert-VITS2 Bert-VITS2 Pytorch 1.2.16 and later Aug 2023
gpt-sovits GPT-SoVITS Pytorch 1.4.0 and later Feb 2024 JP
gpt-sovits-v2 GPT-SoVITS Pytorch 1.4.0 and later Aug 2024

Voice activity detection

Model Reference Exported From Supported Ailia Version Blog
silero-vad Silero VAD Pytorch 1.2.15 and later JP

Voice conversion

Model Reference Exported From Supported Ailia Version Blog
rvc Retrieval-based-Voice-Conversion-WebUI Pytorch 1.2.12 and later JP

Background removal

Model Reference Exported From Supported Ailia Version Blog
U-2-Net U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection Pytorch 1.2.2 and later EN JP
u2net-portrait-matting U^2-Net - Portrait matting Pytorch 1.2.7 and later
u2net-human-seg U^2-Net - human segmentation Pytorch 1.2.4 and later
deep-image-matting Deep Image Matting Keras 1.2.3 and later EN JP
indexnet Indices Matter: Learning to Index for Deep Image Matting Pytorch 1.2.7 and later
modnet MODNet: Trimap-Free Portrait Matting in Real Time Pytorch 1.2.7 and later
background_matting_v2 Real-Time High-Resolution Background Matting Pytorch 1.2.9 and later
cascade_psp CascadePSP Pytorch 1.2.9 and later
rembg Rembg Pytorch 1.2.4 and later
dis_seg Highly Accurate Dichotomous Image Segmentation Pytorch 1.2.10 and later
gfm Bridging Composite and Real: Towards End-to-end Deep Image Matting Pytorch 1.2.10 and later

Crowd counting

Model Reference Exported From Supported Ailia Version Blog
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 EN JP
c-3-framework Crowd Counting Code Framework(C^3-Framework) Pytorch 1.2.5 and later

Deep fashion

Model Reference Exported From Supported Ailia Version Blog
clothing-detection Clothing-Detection Pytorch 1.2.1 and later EN JP
mmfashion MMFashion Pytorch 1.2.5 and later EN JP
mmfashion_tryon MMFashion virtual try-on Pytorch 1.2.8 and later
mmfashion_retrieval MMFashion In-Shop Clothes Retrieval 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
person-attributes-recognition-crossroad person-attributes-recognition-crossroad-0230 Pytorch 1.2.10 and later

Depth estimation

Model Reference Exported From Supported Ailia Version Blog
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 EN JP
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
lap-depth LapDepth-release Pytorch 1.2.9 and later
hitnet ONNX-HITNET-Stereo-Depth-estimation Pytorch 1.2.9 and later
crestereo ONNX-CREStereo-Depth-Estimation Pytorch 1.2.13 and later
mobilestereonet MobileStereoNet Pytorch 1.2.13 and later
zoe_depth ZoeDepth Pytorch 1.3.0 and later
DepthAnything DepthAnything Pytorch 1.2.9 and later

Diffusion

Text to image

Model Reference Exported From Supported Ailia Version Blog
latent-diffusion-txt2img Latent Diffusion - txt2img Pytorch 1.2.10 and later
stable-diffusion-txt2img Stable Diffusion Pytorch 1.2.14 and later JP
control_net ControlNet Pytorch 1.2.15 and later
sd-turbo Hugging Face - SD-Turbo Pytorch 1.2.16 and later
sdxl-turbo Hugging Face - SDXL-Turbo Pytorch 1.2.16 and later
latent-consistency-models latent-consistency-models Pytorch 1.2.16 and later

Text to audio

Model Reference Exported From Supported Ailia Version Blog
riffusion Riffusion Pytorch 1.2.16 and later

Others

Model Reference Exported From Supported Ailia Version Blog
latent-diffusion-inpainting Latent Diffusion - inpainting Pytorch 1.2.10 and later
latent-diffusion-superresolution Latent Diffusion - Super-resolution Pytorch 1.2.10 and later
DA-CLIP DA-CLIP Pytorch 1.2.16 and later
marigold Marigold: Repurposing Diffusion-Based Image Generators for Monocular Depth Estimation Pytorch 1.2.16 and later

Face detection

Model Reference Exported From Supported Ailia Version Blog
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 EN JP
face-mask-detection Face detection using keras-yolov3 Keras 1.2.1 and later EN JP
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 JP
anime-face-detector Anime Face Detector Pytorch 1.2.6 and later
face-detection-adas face-detection-adas-0001 OpenVINO 1.2.5 and later
mtcnn mtcnn Keras 1.2.10 and later

Face identification

Model Reference Exported From Supported Ailia Version Blog
vggface2 VGGFace2 Dataset for Face Recognition Caffe 1.1.0 and later
arcface pytorch implement of arcface Pytorch 1.2.1 and later EN JP
insightface InsightFace: 2D and 3D Face Analysis Project Pytorch 1.2.5 and later
cosface Pytorch implementation of CosFace Pytorch 1.2.10 and later
facenet_pytorch Face Recognition Using Pytorch Pytorch 1.2.6 and later

Face recognition

Age gender estimation

Model Reference Exported From Supported Ailia Version Blog
face_classification Real-time face detection and emotion/gender classification Keras 1.1.0 and later
age-gender-recognition-retail age-gender-recognition-retail-0013 OpenVINO 1.2.5 and later EN JP
mivolo MiVOLO: Multi-input Transformer for Age and Gender Estimation Pytorch 1.2.13 and later

Emotion recognition

Model Reference Exported From Supported Ailia Version Blog
ferplus FER+ CNTK 1.2.2 and later
hsemotion HSEmotion (High-Speed face Emotion recognition) library Pytorch 1.2.5 and later

Gaze estimation

Model Reference Exported From Supported Ailia Version Blog
gazeml A deep learning framework based on Tensorflow
for the training of high performance gaze estimation
TensorFlow 1.2.0 and later
mediapipe_iris irislandmarks.pytorch Pytorch 1.2.2 and later EN JP
ax_gaze_estimation ax Gaze Estimation Pytorch 1.2.2 and later EN JP

Head pose estimation

Model Reference Exported From Supported Ailia Version Blog
hopenet deep-head-pose Pytorch 1.2.2 and later EN JP
6d_repnet 6D Rotation Representation for Unconstrained Head Pose Estimation (Pytorch) Pytorch 1.2.6 and later
L2CS_Net L2CS_Net Pytorch 1.2.9 and later

Keypoint detection

Model Reference Exported From Supported Ailia Version Blog
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 EN JP
prnet Joint 3D Face Reconstruction and Dense Alignment
with Position Map Regression Network
TensorFlow 1.2.2 and later
facemesh facemesh.pytorch Pytorch 1.2.2 and later EN JP
facemesh_v2 MediaPipe Face landmark detection Pytorch 1.2.9 and later JP
3ddfa Towards Fast, Accurate and Stable 3D Dense Face Alignment Pytorch 1.2.10 and later

Others

Model Reference Exported From Supported Ailia Version Blog
ax_facial_features ax Facial Features Pytorch 1.2.5 and later EN
face-anti-spoofing Lightweight Face Anti Spoofing Pytorch 1.2.5 and later EN JP

Face restoration

Model Reference Exported From Supported Ailia Version Blog
gfpgan GFP-GAN: Towards Real-World Blind Face Restoration with Generative Facial Prior Pytorch 1.2.10 and later JP
codeformer CodeFormer: Towards Robust Blind Face Restoration with Codebook Lookup Transformer Pytorch 1.2.9 and later

Face swapping

Model Reference Exported From Supported Ailia Version Blog
sber-swap SberSwap Pytorch 1.2.12 and later JP
facefusion FaceFusion ONNXRuntime 1.2.10 and later

Frame Interpolation

Model Reference Exported From Supported Ailia Version Blog
flavr FLAVR: Flow-Agnostic Video Representations for Fast Frame Interpolation Pytorch 1.2.7 and later EN JP
cain Channel Attention Is All You Need for Video Frame Interpolation Pytorch 1.2.5 and later
film FILM: Frame Interpolation for Large Motion Tensorflow 1.2.10 and later
rife Real-Time Intermediate Flow Estimation for Video Frame Interpolation Pytorch 1.2.13 and later

Generative adversarial networks

Model Reference Exported From Supported Ailia Version Blog
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
restyle-encoder ReStyle Pytorch 1.2.9 and later
sam Age Transformation Using a Style-Based Regression Model Pytorch 1.2.9 and later
encoder4editing Designing an Encoder for StyleGAN Image Manipulation Pytorch 1.2.10 and later
lipgan LipGAN Keras 1.2.15 and later JP
live_portrait LivePortrait Pytorch 1.5.0 and later JP

Hand detection

Model Reference Exported From Supported Ailia Version Blog
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

Hand recognition

Model Reference Exported From Supported Ailia Version Blog
blazehand MediaPipePyTorch Pytorch 1.2.5 and later EN JP
hand3d ColorHandPose3D network TensorFlow 1.2.5 and later
minimal-hand Minimal Hand TensorFlow 1.2.8 and later
v2v-posenet V2V-PoseNet Pytorch 1.2.6 and later
hands_segmentation_pytorch hands-segmentation-pytorch Pytorch 1.2.10 and later

Image captioning

Model Reference Exported From Supported Ailia Version Blog
illustration2vec Illustration2Vec Caffe 1.2.2 and later
image_captioning_pytorch Image Captioning pytorch Pytorch 1.2.5 and later EN JP
blip2 Hugging Face - BLIP-2 Pytorch 1.2.16 and later

Image classification

CNN

Model Reference Exported From Supported Ailia Version Blog
alexnet AlexNet PyTorch Pytorch 1.2.5 and later
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
resnet18 ResNet18 Pytorch 1.2.8 and later
resnet50 Deep Residual Learning for Image Recognition Chainer 1.2.0 and later
wide_resnet50 Wide Resnet Pytorch 1.2.5 and later
inceptionv3 Rethinking the Inception Architecture for Computer Vision Pytorch 1.2.0 and later JP
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
efficientnet A PyTorch implementation of EfficientNet Pytorch 1.2.3 and later
efficientnetv2 EfficientNetV2 Pytorch 1.2.4 and later
mlp_mixer MLP-Mixer Pytorch 1.2.9 and later
convnext A PyTorch implementation of ConvNeXt Pytorch 1.2.5 and later
mobileone A PyTorch implementation of MobileOne Pytorch 1.2.1 and later
imagenet21k ImageNet21K Pytorch 1.2.11 and later
volo VOLO: Vision Outlooker for Visual Recognition Pytorch 1.2.9 and later

Transformer

Model Reference Exported From Supported Ailia Version Blog
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 EN JP
swin-transformer Swin Transformer Pytorch 1.2.6 and later
clip CLIP Pytorch 1.2.9 and later EN JP
japanese-clip Japanese-CLIP Pytorch 1.2.15 and later
japanese-stable-clip-vit-l-16 japanese-stable-clip-vit-l-16 Pytorch 1.2.11 and later

Specific task

Model Reference Exported From Supported Ailia Version Blog
partialconv Partial Convolution Layer for Padding and Image Inpainting Pytorch 1.2.0 and later
weather-prediction-from-image Weather Prediction From Image - (Warmth Of Image) Keras 1.2.5 and later

Image inpainting

Model Reference Exported From Supported Ailia Version Blog
inpainting-with-partial-conv pytorch-inpainting-with-partial-conv PyTorch 1.2.6 and later EN JP
inpainting_gmcnn Image Inpainting via Generative Multi-column Convolutional Neural Networks TensorFlow 1.2.6 and later
3d-photo-inpainting 3D Photography using Context-aware Layered Depth Inpainting Pytorch 1.2.7 and later
deepfillv2 Free-Form Image Inpainting with Gated Convolution Pytorch 1.2.9 and later

Image manipulation

Model Reference Exported From Supported Ailia Version Blog
noise2noise Learning Image Restoration without Clean Data Pytorch 1.2.0 and later
dewarpnet DewarpNet: Single-Image Document Unwarping With Stacked 3D and 2D Regression Networks Pytorch 1.2.1 and later
illnet Document Rectification and Illumination Correction using a Patch-based CNN Pytorch 1.2.2 and later
colorization Colorful Image Colorization Pytorch 1.2.2 and later EN JP
u2net_portrait U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection Pytorch 1.2.2 and later
style2paints Style2Paints TensorFlow 1.2.6 and later
deep_white_balance Deep White-Balance Editing, CVPR 2020 (Oral) PyTorch 1.2.6 and later
deblur_gan DeblurGAN Pytorch 1.2.6 and later
invertible_denoising_network Invertible Image Denoising Pytorch 1.2.8 and later
dfm Deep Feature Matching Pytorch 1.2.6 and later
dfe Deep Fundamental Matrix Estimation Pytorch 1.2.6 and later
dehamer Image Dehazing Transformer with Transmission-Aware 3D Position Embedding Pytorch 1.2.13 and later
pytorch-superpoint pytorch-superpoint : Self-Supervised Interest Point Detection and Description Pytorch 1.2.6 and later
cnngeometric_pytorch CNNGeometric PyTorch implementation Pytorch 1.2.7 and later
lightglue LightGlue-ONNX Pytorch 1.2.15 and later
docshadow DocShadow-ONNX-TensorRT Pytorch 1.2.10 and later
fbcnn Towards Flexible Blind JPEG Artifacts Removal Pytorch 1.2.9 and later

Image restoration

Model Reference Exported From Supported Ailia Version Blog
nafnet NAFNet: Nonlinear Activation Free Network for Image Restoration Pytorch 1.2.10 and later

Image segmentation

Model Reference Exported From Supported Ailia Version Blog
deeplabv3 Xception65 for backbone network of DeepLab v3+ Chainer 1.2.0 and later
hrnet_segmentation High-resolution networks (HRNets) for Semantic Segmentation Pytorch 1.2.1 and later
hair_segmentation hair segmentation in mobile device Keras 1.2.1 and later
pspnet-hair-segmentation pytorch-hair-segmentation Pytorch 1.2.2 and later
human_part_segmentation Self Correction for Human Parsing Pytorch 1.2.4 and later EN JP
semantic-segmentation-mobilenet-v3 Semantic segmentation with MobileNetV3 TensorFlow 1.2.5 and later
pytorch-unet Pytorch-Unet Pytorch 1.2.5 and later
pytorch-enet PyTorch-ENet Pytorch 1.2.8 and later
yet-another-anime-segmenter Yet Another Anime Segmenter Pytorch 1.2.6 and later
swiftnet SwiftNet Pytorch 1.2.6 and later
dense_prediction_transformers Vision Transformers for Dense Prediction Pytorch 1.2.7 and later EN JP
paddleseg PaddleSeg Pytorch 1.2.7 and later EN JP
pp_liteseg PP-LiteSeg Pytorch 1.2.10 and later
suim SUIM Keras 1.2.6 and later
group_vit GroupViT Pytorch 1.2.10 and later
anime-segmentation Anime Segmentation Pytorch 1.2.9 and later
segment-anything Segment Anything Pytorch 1.2.16 and later
tusimple-DUC TuSimple-DUC Pytorch 1.2.10 and later
pytorch-fcn pytorch-fcn Pytorch 1.3.0 and later
grounded_sam Grounded-SAM Pytorch 1.2.16 and later
segment-anything-2 Segment Anything 2 Pytorch 1.2.16 and later
fast_sam FastSAM Pytorch 1.2.14 and later

Large Language Model

Model Reference Exported From Supported Ailia Version Blog
llava LLaVA Pytorch 1.2.16 and later

Landmark classification

Model Reference Exported From Supported Ailia Version Blog
landmarks_classifier_asia Landmarks classifier_asia_V1.1 TensorFlow Hub 1.2.4 and later EN JP
places365 Release of Places365-CNNs Pytorch 1.2.5 and later

Line segment detection

Model Reference Exported From Supported Ailia Version Blog
mlsd M-LSD: Towards Light-weight and Real-time Line Segment Detection TensorFlow 1.2.8 and later EN JP
dexined DexiNed: Dense Extreme Inception Network for Edge Detection Pytorch 1.2.5 and later

Low Light Image Enhancement

Model Reference Exported From Supported Ailia Version Blog
agllnet AGLLNet: Attention Guided Low-light Image Enhancement (IJCV 2021) Pytorch 1.2.9 and later EN JP
drbn_skf DRBN SKF Pytorch 1.2.14 and later

Natural language processing

Bert

Model Reference Exported From Supported Ailia Version Blog
bert pytorch-pretrained-bert Pytorch 1.2.2 and later EN JP
bert_maskedlm huggingface/transformers Pytorch 1.2.5 and later
bert_question_answering huggingface/transformers Pytorch 1.2.5 and later
bert_zero_shot_classification huggingface/transformers Pytorch 1.2.5 and later

Embedding

Model Reference Exported From Supported Ailia Version Blog
sentence_transformers_japanese sentence transformers Pytorch 1.2.7 and later JP
multilingual-e5 multilingual-e5-base Pytorch 1.2.15 and later JP
glucose GLuCoSE (General Luke-based Contrastive Sentence Embedding)-base-Japanese Pytorch 1.2.15 and later

Error corrector

Model Reference Exported From Supported Ailia Version Blog
bert_insert_punctuation bert-japanese Pytorch 1.2.15 and later
t5_whisper_medical error correction of medical terms using t5 Pytorch 1.2.13 and later
bertjsc bertjsc Pytorch 1.2.15 and later

Grapheme to phoneme

Model Reference Exported From Supported Ailia Version Blog
soundchoice-g2p Hugging Face - speechbrain/soundchoice-g2p Pytorch 1.2.16 and later
g2p_en g2p_en Pytorch 1.2.14 and later

Named entity recognition

Model Reference Exported From Supported Ailia Version Blog
bert_ner huggingface/transformers Pytorch 1.2.5 and later
t5_base_japanese_ner t5-japanese Pytorch 1.2.13 and later
bert_ner_japanese jurabi/bert-ner-japanese Pytorch 1.2.10 and later

Reranker

Model Reference Exported From Supported Ailia Version Blog
cross_encoder_mmarco jeffwan/mmarco-mMiniLMv2-L12-H384-v Pytorch 1.2.10 and later JP
japanese-reranker-cross-encoder hotchpotch/japanese-reranker-cross-encoder-large-v1 Pytorch 1.2.16 and later

Sentence generation

Model Reference Exported From Supported Ailia Version Blog
gpt2 GPT-2 Pytorch 1.2.7 and later
rinna_gpt2 japanese-pretrained-models Pytorch 1.2.7 and later

Sentiment analysis

Model Reference Exported From Supported Ailia Version Blog
bert_sentiment_analysis huggingface/transformers Pytorch 1.2.5 and later
bert_tweets_sentiment huggingface/transformers Pytorch 1.2.5 and later

Summarize

Model Reference Exported From Supported Ailia Version Blog
bert_sum_ext BERTSUMEXT Pytorch 1.2.7 and later
presumm PreSumm Pytorch 1.2.8 and later
t5_base_japanese_title_generation t5-japanese Pytorch 1.2.13 and later JP
t5_base_summarization t5-japanese Pytorch 1.2.13 and later

Translation

Model Reference Exported From Supported Ailia Version Blog
fugumt-en-ja Fugu-Machine Translator Pytorch 1.2.9 and later JP
fugumt-ja-en Fugu-Machine Translator Pytorch 1.2.10 abd later

Network intrusion detection

Model Reference Exported From Supported Ailia Version Blog
bert-network-packet-flow-header-payload bert-network-packet-flow-header-payload Pytorch 1.2.10 and later
falcon-adapter-network-packet falcon-adapter-network-packet Pytorch 1.2.10 and later

Neural Rendering

Model Reference Exported From Supported Ailia Version Blog
nerf NeRF: Neural Radiance Fields Tensorflow 1.2.10 and later EN JP
TripoSR TripoSR Pytorch 1.2.6 and later

NSFW detector

Model Reference Exported From Supported Ailia Version Blog
clip-based-nsfw-detector CLIP-based-NSFW-Detector Keras 1.2.10 and later JP

CNN

Model Reference Exported From Supported Ailia Version Blog
yolov1-tiny YOLO: Real-Time Object Detection Darknet 1.1.0 and later JP
yolov2 YOLO: Real-Time Object Detection Pytorch 1.2.0 and later
yolov2-tiny YOLO: Real-Time Object Detection Pytorch 1.2.6 and later
yolov3 YOLO: Real-Time Object Detection ONNX Runtime 1.2.1 and later EN JP
yolov3-tiny YOLO: Real-Time Object Detection ONNX Runtime 1.2.1 and later
yolov4 Pytorch-YOLOv4 Pytorch 1.2.4 and later EN JP
yolov4-tiny Pytorch-YOLOv4 Pytorch 1.2.5 and later
yolov5 yolov5 Pytorch 1.2.5 and later EN JP
yolov6 YOLOV6 Pytorch 1.2.10 and later
yolov7 YOLOv7 Pytorch 1.2.7 and later
yolov8 YOLOv8 Pytorch 1.2.14.1 and later
yolov8-seg YOLOv8 Pytorch 1.2.14.1 and later
yolov9 YOLOv9 Pytorch 1.2.10 and later
yolor yolor Pytorch 1.2.5 and later
yolox YOLOX Pytorch 1.2.6 and later EN JP
yolox-ti-lite edgeai-yolox Pytorch 1.2.9 and later
yolov YOLOV Pytorch 1.2.10 and later
mobilenet_ssd MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch Pytorch 1.2.1 and later EN JP
maskrcnn Mask R-CNN: real-time neural network for object instance segmentation Pytorch 1.2.3 and later
m2det M2Det: A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network Pytorch 1.2.3 and later EN JP
centernet CenterNet : Objects as Points Pytorch 1.2.1 and later EN JP
pedestrian_detection Pedestrian-Detection-on-YOLOv3_Research-and-APP Keras 1.2.1 and later
efficientdet EfficientDet: Scalable and Efficient Object Detection, in PyTorch Pytorch 1.2.6 and later
nanodet NanoDet Pytorch 1.2.6 and later
picodet PP-PicoDet Pytorch 1.2.10 and later
yolact You Only Look At CoefficienTs Pytorch 1.2.6 and later
fastest-det FastestDet Pytorch 1.2.5 and later
poly_yolo Poly YOLO Keras 1.2.6 and later
crowd_det Detection in Crowded Scenes Pytorch 1.2.13 and later
damo_yolo DAMO-YOLO Pytorch 1.2.9 and later

Transformer

Model Reference Exported From Supported Ailia Version Date Blog
glip GLIP Pytorch 1.2.13 and later Dec 2021
dab-detr DAB-DETR Pytorch 1.2.12 and later Jan 2022
detic Detecting Twenty-thousand Classes using Image-level Supervision Pytorch 1.2.10 and later Jan 2022 EN JP
groundingdino Grounding DINO Pytorch 1.2.16 and later Mar 2023 JP

Specific target

Model Reference Exported From Supported Ailia Version Blog
mobile_object_localizer mobile_object_localizer_v1 TensorFlow Hub 1.2.6 and later EN JP
sku110k-densedet SKU110K-DenseDet Pytorch 1.2.9 and later EN JP
traffic-sign-detection Traffic Sign Detection Tensorflow 1.2.10 and later EN JP
footandball FootAndBall: Integrated player and ball detector Pytorch 1.2.0 and later
qrcode_wechatqrcode qrcode_wechatqrcode Caffe 1.2.15 and later
layout_parsing unstructured-inference Pytorch 1.2.9 and later

Object detection 3d

Model Reference Exported From Supported Ailia Version Blog
3d_bbox 3D Bounding Box Estimation Using Deep Learning and Geometry Pytorch 1.2.6 and later
3d-object-detection.pytorch 3d-object-detection.pytorch Pytorch 1.2.8 and later EN JP
mediapipe_objectron MediaPipe Objectron TensorFlow Lite 1.2.5 and later
egonet EgoNet Pytorch 1.2.9 and later
d4lcn D4LCN Pytorch 1.2.9 and later
did_m3d DID M3D Pytorch 1.2.11 and later

Object tracking

Model Reference Exported From Supported Ailia Version Blog
deepsort Deep Sort with PyTorch Pytorch 1.2.3 and later EN JP
person_reid_baseline_pytorch UTS-Person-reID-Practical Pytorch 1.2.6 and later
abd_net Attentive but Diverse Person Re-Identification Pytorch 1.2.7 and later
siam-mot SiamMOT Pytorch 1.2.9 and later
bytetrack ByteTrack Pytorch 1.2.5 and later EN JP 
qd-3dt Monocular Quasi-Dense 3D Object Tracking Pytorch 1.2.11 and later  
strong_sort StrongSORT Pytorch 1.2.15 and later  
centroids-reid On the Unreasonable Effectiveness of Centroids in Image Retrieval Pytorch 1.2.9 and later  
deepsort_vehicle Multi-Camera Live Object Tracking Pytorch 1.2.9 and later

Optical Flow Estimation

Model Reference Exported From Supported Ailia Version Blog
raft RAFT: Recurrent All Pairs Field Transforms for Optical Flow Pytorch 1.2.6 and later EN JP 

Point segmentation

Model Reference Exported From Supported Ailia Version Blog
pointnet_pytorch PointNet.pytorch Pytorch 1.2.6 and later

Pose estimation

Model Reference Exported From Supported Ailia Version Blog
openpose Code repo for realtime multi-person pose estimation in CVPR'17 (Oral) Caffe 1.2.1 and later
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 and later EN JP
pose_resnet Simple Baselines for Human Pose Estimation and Tracking Pytorch 1.2.1 and later EN JP
blazepose MediaPipePyTorch Pytorch 1.2.5 and later
efficientpose Code repo for EfficientPose TensorFlow 1.2.6 and later
movenet Code repo for movenet TensorFlow 1.2.8 and later EN JP
animalpose MMPose - 2D animal pose estimation Pytorch 1.2.7 and later EN JP
mediapipe_holistic MediaPipe Holistic TensorFlow 1.2.9 and later
ap-10k AP-10K Pytorch 1.2.4 and later
posenet PoseNet Pytorch Pytorch 1.2.10 and later
e2pose E2Pose Tensorflow 1.2.5 and later

Pose estimation 3d

Model Reference Exported From Supported Ailia Version Blog
lightweight-human-pose-estimation-3d Real-time 3D multi-person pose estimation demo in PyTorch.
OpenVINO backend can be used for fast inference on CPU.
Pytorch 1.2.1 and later
3d-pose-baseline A simple baseline for 3d human pose estimation in tensorflow.
Presented at ICCV 17.
TensorFlow 1.2.3 and later
pose-hg-3d Towards 3D Human Pose Estimation in the Wild: a Weakly-supervised Approach Pytorch 1.2.6 and later
blazepose-fullbody MediaPipe TensorFlow Lite 1.2.5 and later EN JP
3dmppe_posenet PoseNet of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image" Pytorch 1.2.6 and later
gast A Graph Attention Spatio-temporal Convolutional Networks for 3D Human Pose Estimation in Video (GAST-Net) Pytorch 1.2.7 and later EN JP
mediapipe_pose_world_landmarks MediaPipe Pose real-world 3D coordinates TensorFlow Lite 1.2.10 and later

Road detection

Model Reference Exported From Supported Ailia Version Blog
codes-for-lane-detection Codes-for-Lane-Detection Pytorch 1.2.6 and later EN JP
roneld RONELD-Lane-Detection Pytorch 1.2.6 and later
road-segmentation-adas road-segmentation-adas-0001 OpenVINO 1.2.5 and later
cdnet CDNet Pytorch 1.2.5 and later
lstr LSTR Pytorch 1.2.8 and later
ultra-fast-lane-detection Ultra-Fast-Lane-Detection Pytorch 1.2.6 and later
yolop YOLOP Pytorch 1.2.6 and later
hybridnets HybridNets Pytorch 1.2.6 and later
polylanenet PolyLaneNet Pytorch 1.2.9 and later

Rotation prediction

Model Reference Exported From Supported Ailia Version Blog
rotnet CNNs for predicting the rotation angle of an image to correct its orientation Keras 1.2.1 and later

Style transfer

Model Reference Exported From Supported Ailia Version Blog
adain Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization Pytorch 1.2.1 and later EN JP
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
animeganv2 PyTorch Implementation of AnimeGANv2 Pytorch 1.2.5 and later
pix2pixHD pix2pixHD: High-Resolution Image Synthesis and Semantic Manipulation with Conditional GANs Pytorch 1.2.6 and later
EleGANt EleGANt: Exquisite and Locally Editable GAN for Makeup Transfer Pytorch 1.2.15 and later

Super resolution

Model Reference Exported From Supported Ailia Version Blog
srresnet Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network Pytorch 1.2.0 and later EN JP
edsr Enhanced Deep Residual Networks for Single Image Super-Resolution Pytorch 1.2.6 and later EN JP
han Single Image Super-Resolution via a Holistic Attention Network Pytorch 1.2.6 and later
real-esrgan Real-ESRGAN Pytorch 1.2.9 and later
rcan-it Revisiting RCAN: Improved Training for Image Super-Resolution Pytorch 1.2.10 and later
swinir SwinIR: Image Restoration Using Swin Transformer Pytorch 1.2.12 and later
Hat Hat Pytorch 1.2.6 and later

Text detection

Model Reference Exported From Supported Ailia Version Blog
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

Text recognition

Model Reference Exported From Supported Ailia Version Blog
etl Japanese Character Classification Keras 1.1.0 and later JP
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 EN JP
easyocr Ready-to-use OCR with 80+ supported languages Pytorch 1.2.6 and later
ndlocr_text_recognition NDL OCR Pytorch 1.2.5 and later

Time-Series Forecasting

Model Reference Exported From Supported Ailia Version Blog
informer2020 Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecasting (AAAI'21 Best Paper) Pytorch 1.2.10 and later

Vehicle recognition

Model Reference Exported From Supported Ailia Version Blog
vehicle-attributes-recognition-barrier vehicle-attributes-recognition-barrier-0042 OpenVINO 1.2.5 and later EN JP
vehicle-license-plate-detection-barrier vehicle-license-plate-detection-barrier-0106 OpenVINO 1.2.5 and later

Vision Language Model

Model Reference Exported From Supported Ailia Version Blog
florence2 Hugging Face - microsoft/Florence-2-base Pytorch 1.2.16 and later
qwen2_vl Qwen2-VL Pytorch 1.5.0 and later

Commercial model

Model Reference Exported From Supported Ailia Version Blog
acculus-pose Acculus, Inc. Caffe 1.2.3 and later

Other languages