Giter VIP home page Giter VIP logo

ailia-models-cpp's Introduction

ailia-models-cpp

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

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, Android and Jetson. 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.

Install ailia SDK

Download ailia SDK

The ailia SDK can be installed from a git submodule.

git subomdule init
git submodule update

Download license file

Download the ailia SDK license file with the following command. The license file is valid for one month.

cd ailia
python3 download_license.py

Alternatively, the license file can be obtained by requesting the evaluation version of the ailia SDK.

Request trial version

Install dependent libraries

Windows

cmake, opencv and Visual Studio 2019 or newer are required.

Please set the OpenCV path to the environment variable.

  • OpenCV_DIR: C:\opencv\build

Mac

Xcode Commandline Tools are required, they can be installed by running the command.

xcode-select --install

cmake and OpenCV is required, it can be installed by running the command.

brew install cmake
brew install opencv

Linux

cmake and OpenCV is required, it can be installed by running the command.

apt install cmake
apt install libopencv-dev

Build

Please run the following command in the root folder of your local repository.

cmake .
cmake --build .

Run

Move to the model folder, execute sh or bat, then the model file will be downloaded and the model will run.

Windows

In the case of Windows, please copy dll files such as ailia.dll to the execution directory.

cd object_detection/yolox
./yolox.bat

Mac and Linux

cd object_detection/yolox
./yolox.sh

You can use the web camera by adding the -v 0 option.

cd object_detection/yolox
./yolox.sh -v 0

Supporting Models

Audio processing

Name Detail Exported From Supported Ailia Version
silero-vad Silero VAD Pytorch 1.2.15 and later
clap CLAP Pytorch 1.3.0 and later
whisper Whisper Pytorch 1.2.16 and later

Background removal

Name Detail Exported From Supported Ailia Version
U-2-Net U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection Pytorch 1.2.2 and later

Image classification

Name Detail Exported From Supported Ailia Version
resnet50 Deep Residual Learning for Image Recognition Chainer 1.2.0 and later
clip CLIP Pytorch 1.2.9 and later

Face detection

Name Detail Exported From Supported Ailia Version
yolov3-face YOLO: Real-Time Object Detection ONNX Runtime 1.2.1 and later

Face identification

Name Detail Exported From Supported Ailia Version
arcface pytorch implement of arcface Pytorch 1.2.1 and later

Face recognition

Name Detail Exported From Supported Ailia Version
face_alignment 2D and 3D Face alignment library build using pytorch Pytorch 1.2.1 and later
mediapipe_iris irislandmarks.pytorch Pytorch 1.2.2 and later

Natural language processing

Name Detail Exported From Supported Ailia Version
fugumt-en-ja Fugu-Machine Translator Pytorch 1.2.9 and later
fugumt-ja-en Fugu-Machine Translator Pytorch 1.2.10 and later
bert_maskedlm huggingface/transformers Pytorch 1.2.5 and later
sentence_transformers sentence transformers Pytorch 1.2.7 and later
t5_whisper_medical error correction of medical terms using t5 Pytorch 1.2.13 and later
multilingual-e5 multilingual-e5-base Pytorch 1.2.15 and later

Object detection

Name Detail Exported From Supported Ailia Version
yolov3-tiny YOLO: Real-Time Object Detection ONNX Runtime 1.2.1 and later
m2det M2Det: A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network Pytorch 1.2.3 and later
yolox YOLOX Pytorch 1.2.6 and later

Pose estimation

Name Detail Exported From Supported Ailia Version
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

Other languages

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.