
onnx
心惠天意
一只展翅的小菜鸟
展开
-
onnx篇---onnx转换成rknn
多张图预测# -- coding: utf-8 --import osimport urllibimport tracebackimport timeimport sysimport numpy as npimport cv2from matplotlib import pyplot as pltfrom rknn.api import RKNN def show_outputs(outputs): # output = outputs[0][0] output =原创 2021-08-17 13:09:17 · 5768 阅读 · 1 评论 -
onnx篇---验证onnx转换成功与否
1. 调用opencv 来查看网路输出的结果use_opencv.pyimport cv2import onnxruntimeimport numpy as nponnx_path_demo = '../data/weights/Yolo_bin.onnx'# onnx_path_demo = 'YOLO.onnx'image_path = '../data/img/0601_Bin_061.jpg'image_src = cv2.imread(image_path)resized原创 2021-08-17 11:56:18 · 3236 阅读 · 0 评论