pynq-onnx-demo's Introduction

pynq-onnx-demo

在pynq上的onnx神经网络推理示例

使用说明

data: 存储测试数据，带k的为keras输入模式(N, L, C),不带k的为pytorch输入模式(N, C, L).其中C=8，L=2048
model_files:模型文件。.pth和.h5为训练得到的pytorch和keras的模型文件；带k的onnx文件由h5转化而来，不带k的由pth转化而来
models: Alexnet1d的pytorch代码
keras2onnx: keras .h5模型文件转化为onnx
torch2onnx: pytorch .pth模型文件转化为onnx
infer_onnx_from_keras: 运行由keras模型文件转化而来的onnx
infer_onnx_from_torch: 运行由pytorch模型文件转化而来的onnx

整体工程可放置于PYNQ环境下，

运行infer_onnx_from_torch.py 或 infer_onnx_from_keras.py观察推理速度。

Recommend Projects

React

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

Vue.js

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

Typescript

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

TensorFlow

An Open Source Machine Learning Framework for Everyone

Django

The Web framework for perfectionists with deadlines.

Laravel

A PHP framework for web artisans

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.

Visualization

Some thing interesting about visualization, use data art

Game

Some thing interesting about game, make everyone happy.

Recommend Org

Facebook

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

Microsoft

Open source projects and samples from Microsoft.

Google

Google ❤️ Open Source for everyone.

Alibaba

Alibaba Open Source for everyone

D3

Data-Driven Documents codes.

Tencent

China tencent open source team.