Easy-to-use UI for automatically sparsifying neural networks and creating sparsification recipes for better inference performance and a smaller footprint

✭ 138

javascript python computer-vision tensorflow keras pytorch image-classification pruning object-detection quantization automl onnx deep-learning-accelerator inference-performance sparsification-recipe smaller-models sparsify

onnx2tensorRt

tensorRt-inference darknet2onnx pytorch2onnx mxnet2onnx python version

✭ 14

python mxnet pytorch darknet tensorrt onnx yolov4

ONNX-Mobile-Human-Pose-3D

Python scripts for performing 3D human pose estimation using the Mobile Human Pose model in ONNX.

✭ 69

python pose-estimation onnx 3d-pose-estimation onnxruntime yolov5s

ONNX-HITNET-Stereo-Depth-estimation

Python scripts form performing stereo depth estimation using the HITNET model in ONNX.

✭ 21

python deep-learning stereo-vision stereo-matching depth-estimation onnx onnxruntime stereo-depth-estimation

serving-runtime

Exposes a serialized machine learning model through a HTTP API.

✭ 15

java CSS shell rust python HTML machine-learning tensorflow inference hdf5 serving onnx

finn-base

Open Source Compiler Framework using ONNX as Frontend and IR

✭ 21

python Verilog shell compiler intermediate-representation onnx finn

ONNX-ImageNet-1K-Object-Detector

Python scripts for performing object detection with the 1000 labels of the ImageNet dataset in ONNX. The repository combines a class agnostic object localizer to first detect the objects in the image, and next a ResNet50 model trained on ImageNet is used to label each box.

✭ 18

python opencv computer-vision imagenet object-detection onnx object-localization class-agnostic-detection

ai-deployment

关注AI模型上线、模型部署

✭ 149

Jupyter Notebook mxnet tensorflow deploy scikit-learn keras pytorch pyspark xgboost lightgbm pmml spark-ml onnx

AI-LAB

This repository contains a docker image that I use to develop my artificial intelligence applications in an uncomplicated fashion. Python, TensorFlow, PyTorch, ONNX, Keras, OpenCV, TensorRT, Numpy, Jupyter notebook... 🐋🔥

✭ 44

Dockerfile docker dockerfile data-science machine-learning ai computer-vision deep-learning ubuntu tensorflow jupyter-notebook pytorch artificial-intelligence tensorrt onnx

person-detection

TensorRT person tracking RFBNet300

✭ 30

C++CMake Cuda c detection tensorrt tracking-by-detection onnx

ONNX-Runtime-with-TensorRT-and-OpenVINO

Docker scripts for building ONNX Runtime with TensorRT and OpenVINO in manylinux environment

✭ 15

C++python shell Cuda cuda nvidia tensorrt onnx openvino onnxruntime

model-zoo-old

The ONNX Model Zoo is a collection of pre-trained models for state of the art models in deep learning, available in the ONNX format

✭ 38

Jupyter Notebook python deep-learning mxnet models cnn neural-networks gluon onnx

openvino pytorch layers

How to export PyTorch models with unsupported layers to ONNX and then to Intel OpenVINO

✭ 17

C++python CMake custom intel pytorch fft onnx unpooling openvino model-optimizer intel-openvino

wonnx

A GPU-accelerated ONNX inference run-time written 100% in Rust, ready for the web

✭ 160

rust python webgpu onnx

Fast Stacked Hourglass Network OpenVino

A fast stacked hourglass network for human pose estimation on OpenVino

✭ 52

python mpii-dataset pose-estimation ncs mobilenet stacked-hourglass-networks onnx openvino

YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

✭ 6,570

python C++shell deep-learning pytorch yolo object-detection tensorrt ncnn onnx yolov3 openvino megengine yolox

lego-art-remix

Powerful computer vision assisted Lego mosaic creator · Over 500,000 images created (so far!)

✭ 148

javascript HTML computer-vision lego onnx deep-neural-network lego-mosaic lego-art custom-pictures lego-art-remix

vs-mlrt

Efficient ML Filter Runtimes for VapourSynth (with built-in support for waifu2x, DPIR, RealESRGANv2, and Real-CUGAN)

✭ 34

C++python CMake ai deep-learning neural-network gpu cuda ml waifu2x artificial-intelligence vapoursynth tensorrt onnx openvino onnxruntime real-esrgan dpir real-cugan

optimum

🏎️ Accelerate training and inference of 🤗 Transformers with easy to use hardware optimization tools

✭ 567

python Makefile training optimization intel transformers inference pytorch quantization onnx onnxruntime graphcore habana

yolov5 for rknn

YOLOv5 in PyTorch > ONNX > RKNN

✭ 79

pytorch onnx yolov5 rknn

popart

Poplar Advanced Runtime for the IPU

✭ 62

C++python CMake PHP shell Jinja machine-learning ai deep-learning ipu onnx graphcore popart

EfficientIR

人工智障本地图片检索工具 | An EfficientNet based image retrieval tool

✭ 64

python image-search reverse-image-search image-retrieval similarity-search onnx hnswlib efficientnet-pytorch

SynapseML

Simple and Distributed Machine Learning

ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

✭ 281

python Cuda C++c open-source caffe deep-learning neural-network cuda pytorch quantization onnx

Unity Detection2AR

Localize 2D image object detection in 3D Scene with Yolo in Unity Barracuda and ARFoundation.

✭ 147

C#unity augmented-reality unity3d object-detection yolov2 barracuda onnx yolov3 object-localization arfoundation

yolov5 onnx2caffe

yolov5 onnx caffe

✭ 73

python caffe onnx yolov5

ngraph-onnx

nGraph™ Backend for ONNX

✭ 42

python groovy shell Dockerfile ngraph onnx onnx-support onnx-backend

yolov5 tensorrt int8 tools

tensorrt int8 量化yolov5 onnx模型

✭ 105

python tensorrt int8 onnx yolov5

YOLOv4MLNet

Use the YOLO v4 and v5 (ONNX) models for object detection in C# using ML.Net

✭ 61

C#machine-learning computer-vision neural-network dotnet ml dotnet-core yolo v4 v5 onnx ml-net yolov4 yolov5

tractjs

Run ONNX and TensorFlow inference in the browser.

✭ 67

rust typescript javascript machine-learning browser wasm neural-networks onnx tensorflow-inference

onnxruntime-rs

Rust wrapper for Microsoft's ONNX Runtime (version 1.8)

✭ 149

rust inference rust-wrapper onnx onnxruntime onnx-runtime onnxruntime-sys

gluon2pytorch

Gluon to PyTorch deep neural network model converter

✭ 72

python converter machine-learning deep-neural-networks deep-learning mxnet pytorch densenet resnet gluon darknet dual-path-networks onnx densenet-pytorch nasnet pytorch-converter dualpath-network

mtomo

Multiple types of NN model optimization environments. It is possible to directly access the host PC GUI and the camera to verify the operation. Intel iHD GPU (iGPU) support. NVIDIA GPU (dGPU) support.

✭ 24

Dockerfile shell mxnet tensorflow models pytorch tensorrt myriad onnx tensorflowlite tflite tensorflowjs openvino models-converter edgetpu depthai tf-trt

AnimeGANv2-ONNX-Sample

「PyTorch Implementation of AnimeGANv2」のPythonでのONNX推論サンプル

✭ 54

Jupyter Notebook python pytorch onnx colaboratory animeganv2 onnxruntime-gpu

pytorch-android

[EXPERIMENTAL] Demo of using PyTorch 1.0 inside an Android app. Test with your own deep neural network such as ResNet18/SqueezeNet/MobileNet v2 and a phone camera.

✭ 105