cqu20160901/yolov8n_onnx_tensorRT_rknn_horizon cqu20160901/yolov8n_onnx_tensorRT_rknn_horizon

  • Stars
    star
    127
  • Rank 282,790 (Top 6 %)
  • Language
    Python
  • Created almost 2 years ago
  • Updated 11 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
cqu20160901/yolov8n_onnx_tensorRT_rknn_horizon
github

cqu20160901

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

yolov8n 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（onnx、tensorRT、RKNN、Horzion）

yolov8n_onnx_tensorRT_rknn_horizon

yolov8n 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。

文件夹结构说明

yolov8n_onnx：onnx模型、测试图像、测试结果、测试demo脚本

yolov8n_TensorRT：TensorRT版本模型、测试图像、测试结果、测试demo脚本、onnx模型、onnx2tensorRT脚本(tensorRT-7.2.3.4)

yolov8n_rknn：rknn模型、测试（量化）图像、测试结果、onnx2rknn转换测试脚本

yolov8n_horizon：地平线模型、测试（量化）图像、测试结果、转换测试脚本、测试量化后onnx模型脚本

测试结果

image

（注：图片来源coco128）

说明：推理测试预处理没有考虑等比率缩放，激活函数 SiLU 用 Relu 进行了替换。由于使用的是coco128数据进行训练的，且迭代的次数不多，效果并不是很好，仅供测试流程用。

导出onnx参考 yolov8 瑞芯微RKNN和地平线Horizon芯片仿真测试部署-2023年11月15日版本

导出onnx参考 yolov8 瑞芯微RKNN和地平线Horizon芯片仿真测试部署-2023年2月版本

该种方式优缺点

优点：板端部署通用性好，在板端效率高。 缺点：工程难道大。

官方导出onnx方式板端部署方式参考 官方导出onnx方式部署 。

板端C++部署（以rknn芯片为例）

rknn的板端部署参考 C++部署

yolov8seg 部署参考

想试试yolov8seg的小伙伴看过来：模型+代码

More Repositories

1

yolov8_rknn_Cplusplus

yolov8 瑞芯微 rknn 板端 C++部署。
C
88
star
2

yolov10_rknn_Cplusplus

yolov10 瑞芯微 rknn 板端 C++部署，使用平台 rk3588。
C
54
star
3

yolov10_onnx_rknn_horizon_tensorRT

yolov10 目标检测部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网部署最简单、运行速度最快的部署方式（全网首发）。
Python
39
star
4

yolov8pose_onnx_tensorRT_rknn_horizon

yolov8pose 部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。
Python
36
star
5

yolov8n_official_onnx_tensorRT_rknn_horizon

yolov8n 部署版，基于官方的导出onnx脚本导出onnx模型，在不同平台上进行部署测试，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。
Python
32
star
6

yolov7_caffe_onnx_tensorRT_rknn_horizon

yolov7 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（caffe、onnx、tensorRT、RKNN、Horzion）。
Python
31
star
7

yolov8pose_rknn_Cplusplus

yolov8pose 瑞芯微 rknn 板端 C++部署。
C
29
star
8

yolov8n_onnx_tensorRT_rknn_horizon_dfl

yolov8n 目标检测部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网部署最简单、速度最快的部署方式。
Python
26
star
9

yolov5p6_caffe_onnx_tensorRT_rknn_horizon

yolov5_nano_p6 部署版本，后处理用python语言和C形式进行改写，便于移植不同平台caffe、onnx、tensorRT、rknn、horizon。
Python
24
star
10

yolov8seg_onnx_tensorRT_rknn_horizon

yolov8seg 部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。
Python
23
star
11

yolov8n_rknn_Cplusplus_dfl

yolov8 瑞芯微 rknn 板端 C++部署，使用平台 rk3588，全网最简单、运行最快的部署方式。
C
21
star
12

yolov8-obb_onnx_rknn_horizon_tensonRT

yolov8 旋转目标检测部署，瑞芯微RKNN芯片部署、地平线Horizon芯片部署、TensorRT部署
Python
18
star
13

UNetMultiLane_onnx_tensorRT_rknn_horizon

UNetMultiLane 多车道线和车道线类型识别部署版本，测试不同平台部署（onnx、tensorRT、RKNN、Horzion），可识别所在的车道和车道线的类型。
Python
17
star
14

yolov8seg_rknn_Cplusplus

yolov8seg 瑞芯微 rknn 板端 C++部署，使用平台 rk3588。
C
17
star
15

yolop_onnx_tensorRT_rknn_horizon

yolop 部署版本，后处理用python语言以C++方式形式进行改写，便于移植不同平台（onnx、tensorRT、rknn）。
Python
12
star
16

Ultra-Fast-Lane-Detection_caffe_onnx_horizon_rknn

Ultra-Fast-Lane-Detection 部署版本，后处理用进行改写，便于移植不同平台（caffe、onnx、RKNN、Horzion）。
Python
12
star
17

yolov9_onnx_tensorRT_rknn_horizon

yolov9 部署，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网首个通用部署示例。
Python
11
star
18

yolov8_tensorRT_Cplusplus

yolov8 tensorRT 的 C++部署。
C++
11
star
19

yolov9_rknn_Cplusplus

yolov9 瑞芯微 rknn 板端 C++部署。
C
10
star
20

DETR_onnx_tensorRT_V2

DETR tensor去除推理过程无用辅助头+fp16部署再次加速+解决转tensorrt 输出全为0问题的新方法。
Python
10
star
21

FastSAM_onnx_rknn

FastSAM 部署版本，便于移植不同平，部署简单、运行速度快。
Python
10
star
22

yolov8obb_rknn_Cplusplus

yolov8obb 旋转目标检测部署rknn的C++代码
C
9
star
23

yoloworld-tensorRT-Cplusplus

C++
7
star
24

yoloworld-rknn-Cplusplus

C
7
star
25

yolov6n_v2.0_caffe_onnx_rknn

yolov6n 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（caffe、onnx、rknn）
Python
6
star
26

yolov8seg_dfl_rknn_Cplusplus

yolov8seg 瑞芯微 rknn 板端 C++部署，使用平台 rk3588，部署难度小，运行速度快。
C
6
star
27

yolov8seg_onn_rknn_horizon_tensorRT_dfl

yolov8seg 部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网部署最简单、速度最快的部署方式。
Python
6
star
28

yoloworld-onnx-tensorRT-rknn-horizon

yoloworld 的onnx、tensorRT、rknn、horizon 部署，通用各种平台和芯片。
Python
6
star
29

DETR_tensorRT_Cplusplus

DETR tensorRT C++部署。
C++
6
star
30

MiDaS_DepthEstimation_rknn_Cplusplus

MiDaS（Multiple Depth Estimation Accuracy with Single Network）单目深度估计，部署 rk3588。
C
6
star
31

DETR_onnx_tensorRT

DETR tensorRT 部署。
Python
6
star
32

UNetMultiLane

UNetMultiLane 多车道线、车道线类型识别，自我学习使用，没有经过大量测试，不免有问题，不喜不扰。
Python
6
star
33

UNetMultiLane_rknn_Cplusplus

UNetMultiLane 多车道线和车道线类型识别rknn部署C++代码
C
5
star
34

yolov6n_caffe_onnx_tensorRT

yolov6n 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（caffe、onnx、tensorRT）。
Python
5
star
35

centernet3d_onnx_rknn_horizon_tensorRT

CenterNet3D 部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。
Python
5
star
36

ssd_caffe_onnx_tensorRT

用python语言以C++的风格重写SSD的后处理，caffe、onnx、tensorRT三个版本
Python
4
star
37

yolox_caffe_onnx_tensorRT

yolox 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（caffe、onnx、tensorRT）。
Python
4
star
38

cqu20160901

4
star
39

plateRegress_caffe_onnx_tensorRT

车牌回归，caffe、onnx、tensorRT版本，不涉及训练过程，只涉及前向推理和移植
Python
4
star
40

yolov8pose_dfl_rknn_Cplusplus

yolov8pose 部署版本，将DFL放在后处理中，部署rk3588.
C
4
star
41

yolov8pose_onn_rknn_horizon_tensorRT_dfl

yolov8pose 部署版本，将DFL放在后处理中，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网部署最简单、速度最快的部署方式。
Python
4
star
42

CenterNet_rknn_Cplusplus

centernet 瑞芯微 rknn 板端 C++部署。
C
4
star
43

FastSAM_rknn_Cplusplus

FastSAM 部署rknn C++ 代码
C
3
star
44

yolov9_tensorRT_Cplusplus

yolov9 tensorRT C++部署。
C++
3
star
45

FCOS_onnx

FCOS_onnx用python语言和C语言的风格进行进行了重写，便于不同平台进行移植。
Python
3
star
46

DeepLabV3_pytorch_onnx_tensorRT

路面分割，pytorch、onnx、tensorRT版本，不涉及训练过程，只涉及前向推理移植
Python
3
star
47

RFB_Face_caffe_onnx_tensorRT

RFB320_Face_Detector 部署版本（不涉及训练），后处理用python语言和C++语言形式进行改写，便于移植不同平台（caffe、onnx、tensorRT）。
Python
3
star
48

CenterNet_onnx_rknn_horizon_tensorRT

CenterNet 部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon）。本来想做基于CenterNet的单目3D目标检测，但觉得要先把centenet后处理手撸一遍，说干就干。
Python
3
star
49

LPRNet_rknn_Cplusplus

车牌识别 LPRNet : License Plate Recognition via Deep Neural Networks 部署rk3588，C++实现。
C
2
star
50

yolov11_tensorRT_postprocess_cuda

yolov11 的tensorRT C++ 部署，后处理使用cuda进行耗时操作。
C++
1
star
51

RT-DETRv2_TensorRT_Cplusplus

RT-DETRv2 tensorrt C++ 部署
C++
1
star