Skip to main content

ncnn is a high-performance neural network inference framework optimized for the mobile platform

Project description

ncnn-vulkan

This fork aims to provide vulkan functionality to the ncnn python bindings, which is absent in the official package. All that is needed to use vulkan is the addition of net.opt.use_vulkan_compute = True.

Also in this fork, failed vram allocation will result in a runtime error that is catchable, allowing you to clear the allocated vram on this error by using ncnn.destroy_gpu_instance().

Minimal example:

import cv2
from ncnn_vulkan import ncnn
import numpy as np

net = ncnn.Net()

# Use vulkan compute
net.opt.use_vulkan_compute = True

# Load model param and bin
net.load_param("./x4.param")
net.load_model("./x4.bin")

ex = net.create_extractor()

# Load image using opencv
img = cv2.imread("./example.jpg")

# Convert image to ncnn Mat
mat_in = ncnn.Mat.from_pixels(
    img,
    ncnn.Mat.PixelType.PIXEL_BGR,
    img.shape[1],
    img.shape[0]
)

# Normalize image (required)
# Note that passing in a normalized numpy array will not work.
mean_vals = []
norm_vals = [1 / 255.0, 1 / 255.0, 1 / 255.0]
mat_in.substract_mean_normalize(mean_vals, norm_vals)

# Try/except block to catch out-of-memory error
try:
    # Make sure the input and output names match the param file
    ex.input("data", mat_in)
    ret, mat_out = ex.extract("output")
    out = np.array(mat_out)

    # Transpose the output from `c, h, w` to `h, w, c` and put it back in 0-255 range
    output = out.transpose(1, 2, 0) * 255

    # Save image using opencv
    cv2.imwrite('./out.png', output)
except:
    ncnn.destroy_gpu_instance()

original readme

License download codecov Language grade: C/C++

ncnn is a high-performance neural network inference computing framework optimized for mobile platforms. ncnn is deeply considerate about deployment and uses on mobile phones from the beginning of design. ncnn does not have third party dependencies. It is cross-platform, and runs faster than all known open source frameworks on mobile phone cpu. Developers can easily deploy deep learning algorithm models to the mobile platform by using efficient ncnn implementation, create intelligent APPs, and bring the artificial intelligence to your fingertips. ncnn is currently being used in many Tencent applications, such as QQ, Qzone, WeChat, Pitu and so on.

ncnn 是一个为手机端极致优化的高性能神经网络前向计算框架。ncnn 从设计之初深刻考虑手机端的部署和使用。无第三方依赖,跨平台,手机端 cpu 的速度快于目前所有已知的开源框架。基于 ncnn,开发者能够将深度学习算法轻松移植到手机端高效执行,开发出人工智能 APP,将 AI 带到你的指尖。ncnn 目前已在腾讯多款应用中使用,如 QQ,Qzone,微信,天天P图等。


技术交流QQ群:637093648(超多大佬) 答案:卷卷卷卷卷 (已满)

Pocky QQ群(MLIR YES!): 677104663(超多大佬) 答案:multi-level intermediate representation

Telegram Group https://t.me/ncnnyes

Discord Channel https://discord.gg/YRsxgmF


Current building status matrix

System CPU (32bit) CPU (64bit) GPU (32bit) GPU (64bit)
Linux (GCC) Build Status Build Status Build Status
Linux (Clang) Build Status Build Status Build Status
Linux (ARM) Build Status Build Status
Linux (MIPS) Build Status Build Status
Linux (RISC-V) Build Status
Linux (LoongArch) Build Status
Windows Build Status Build Status Build Status
Windows (ARM) Build Status Build Status
macOS Build Status Build Status
macOS (ARM) Build Status Build Status
Android Build Status Build Status Build Status Build Status
Android-x86 Build Status Build Status Build Status Build Status
iOS Build Status Build Status Build Status
iOS Simulator Build Status Build Status
WebAssembly Build Status
RISC-V GCC/Newlib Build Status Build Status

Support most commonly used CNN network

支持大部分常用的 CNN 网络


HowTo

how to build ncnn library on Linux / Windows / macOS / Raspberry Pi3 / Android / NVIDIA Jetson / iOS / WebAssembly / AllWinner D1 / Loongson 2K1000

download prebuild binary package for android and ios

use ncnn with alexnet with detailed steps, recommended for beginners :)

ncnn 组件使用指北 alexnet 附带详细步骤,新人强烈推荐 :)

use netron for ncnn model visualization

out-of-the-box web model conversion

ncnn low-level operation api

ncnn param and model file spec

ncnn operation param weight table

how to implement custom layer step by step


FAQ

ncnn throw error

ncnn produce wrong result

ncnn vulkan


Features

  • Supports convolutional neural networks, supports multiple input and multi-branch structure, can calculate part of the branch
  • No third-party library dependencies, does not rely on BLAS / NNPACK or any other computing framework
  • Pure C++ implementation, cross-platform, supports android, ios and so on
  • ARM NEON assembly level of careful optimization, calculation speed is extremely high
  • Sophisticated memory management and data structure design, very low memory footprint
  • Supports multi-core parallel computing acceleration, ARM big.LITTLE cpu scheduling optimization
  • Supports GPU acceleration via the next-generation low-overhead vulkan api
  • Extensible model design, supports 8bit quantization and half-precision floating point storage, can import caffe/pytorch/mxnet/onnx/darknet/keras/tensorflow(mlir) models
  • Support direct memory zero copy reference load network model
  • Can be registered with custom layer implementation and extended
  • Well, it is strong, not afraid of being stuffed with 卷 QvQ

功能概述

  • 支持卷积神经网络,支持多输入和多分支结构,可计算部分分支
  • 无任何第三方库依赖,不依赖 BLAS/NNPACK 等计算框架
  • 纯 C++ 实现,跨平台,支持 android ios 等
  • ARM NEON 汇编级良心优化,计算速度极快
  • 精细的内存管理和数据结构设计,内存占用极低
  • 支持多核并行计算加速,ARM big.LITTLE cpu 调度优化
  • 支持基于全新低消耗的 vulkan api GPU 加速
  • 可扩展的模型设计,支持 8bit 量化 和半精度浮点存储,可导入 caffe/pytorch/mxnet/onnx/darknet/keras/tensorflow(mlir) 模型
  • 支持直接内存零拷贝引用加载网络模型
  • 可注册自定义层实现并扩展
  • 恩,很强就是了,不怕被塞卷 QvQ

supported platform matrix

  • ✅ = known work and runs fast with good optimization
  • ✔️ = known work, but speed may not be fast enough
  • ❔ = shall work, not confirmed
  • / = not applied
Windows Linux Android macOS iOS
intel-cpu ✔️ ✔️ ✔️ /
intel-gpu ✔️ ✔️ /
amd-cpu ✔️ ✔️ ✔️ /
amd-gpu ✔️ ✔️ /
nvidia-gpu ✔️ ✔️ /
qcom-cpu ✔️ / /
qcom-gpu ✔️ ✔️ / /
arm-cpu / /
arm-gpu ✔️ / /
apple-cpu / / / ✔️
apple-gpu / / / ✔️ ✔️

Example project


License

BSD 3 Clause

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

ncnn_vulkan-2023.6.18-cp311-cp311-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.11 Windows x86-64

ncnn_vulkan-2023.6.18-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.11 manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp311-cp311-macosx_11_0_arm64.whl (7.0 MB view hashes)

Uploaded CPython 3.11 macOS 11.0+ ARM64

ncnn_vulkan-2023.6.18-cp311-cp311-macosx_10_9_x86_64.whl (8.8 MB view hashes)

Uploaded CPython 3.11 macOS 10.9+ x86-64

ncnn_vulkan-2023.6.18-cp310-cp310-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.10 Windows x86-64

ncnn_vulkan-2023.6.18-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp310-cp310-macosx_11_0_arm64.whl (7.0 MB view hashes)

Uploaded CPython 3.10 macOS 11.0+ ARM64

ncnn_vulkan-2023.6.18-cp310-cp310-macosx_10_9_x86_64.whl (9.0 MB view hashes)

Uploaded CPython 3.10 macOS 10.9+ x86-64

ncnn_vulkan-2023.6.18-cp39-cp39-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.9 Windows x86-64

ncnn_vulkan-2023.6.18-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp39-cp39-macosx_11_0_arm64.whl (7.0 MB view hashes)

Uploaded CPython 3.9 macOS 11.0+ ARM64

ncnn_vulkan-2023.6.18-cp39-cp39-macosx_10_9_x86_64.whl (8.8 MB view hashes)

Uploaded CPython 3.9 macOS 10.9+ x86-64

ncnn_vulkan-2023.6.18-cp38-cp38-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.8 Windows x86-64

ncnn_vulkan-2023.6.18-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp38-cp38-macosx_11_0_arm64.whl (7.0 MB view hashes)

Uploaded CPython 3.8 macOS 11.0+ ARM64

ncnn_vulkan-2023.6.18-cp38-cp38-macosx_10_9_x86_64.whl (8.8 MB view hashes)

Uploaded CPython 3.8 macOS 10.9+ x86-64

ncnn_vulkan-2023.6.18-cp37-cp37m-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.7m Windows x86-64

ncnn_vulkan-2023.6.18-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.7m manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp37-cp37m-macosx_10_9_x86_64.whl (8.8 MB view hashes)

Uploaded CPython 3.7m macOS 10.9+ x86-64

ncnn_vulkan-2023.6.18-cp36-cp36m-win_amd64.whl (2.8 MB view hashes)

Uploaded CPython 3.6m Windows x86-64

ncnn_vulkan-2023.6.18-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB view hashes)

Uploaded CPython 3.6m manylinux: glibc 2.17+ x86-64

ncnn_vulkan-2023.6.18-cp36-cp36m-macosx_10_9_x86_64.whl (8.8 MB view hashes)

Uploaded CPython 3.6m macOS 10.9+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page