首页>源码>c++语言>yolov5s-jtbz-npu

yolov5s-jtbz-npu

声明:资源链接索引至第三方,平台不作任何存储,仅提供信息检索服务,若有版权问题,请https://help.coders100.com提交工单反馈
在KhadasVIM3上使用tengine部署YOLOv5s交通标志识别模型到NPU,平均推理速度为49.83ms。以下是详细的描述:

1. 首先,需要在KhadasVIM3上安装tengine和YOLOv5s模型。可以通过以下命令进行安装:

pip install tengine
pip install opencv-python
pip install yolov5s


2. 下载YOLOv5s模型,并将其解压到一个文件夹中。例如,将文件放在`/home/user/yolov5s`目录下。

3. 修改YOLOv5s配置文件,设置输入图像大小、输出类别数量等参数。例如,将输入图像大小设置为`640x640`,输出类别数量设置为`10`。

4. 编写Python脚本,使用tengine引擎执行YOLOv5s模型。在脚本中,需要导入所需的库并调用tengine的API。例如:

import tengine
from tengine import TEngine
from yolo_utils.yolo_model import YoloModel

# 加载模型
model = YoloModel(config_file='yolov5s.cfg', weights='yolov5s.weights')

# 创建tengine实例
tengine = TEngine(host='localhost', port=7000, model=model)

# 准备输入数据
input_image = cv2.imread('/path/to/image')
input_image = cv2.resize(input_image, (640, 640))

# 开始推理
output = tengine.predict([input_image], output_dict=True)

# 处理输出结果
for key, value in output.items():
if key == 'score':
print('{}: {:.4f}'.format(value['class_id'], value['score']))


5. 运行Python脚本,观察输出结果。根据输出结果,可以对模型进行调整以获得更好的性能。

6. 将训练好的模型上传到KhadasVIM3服务器,并使用tengine引擎执行模型。这样,交通标志识别任务就可以在NPU上高效地进行推理。复现在KhadasVIM3上使用tengine部署yolov5s的交通标志识别模型到NPU上,平均推理速度49.83ms
电信网络下载

访问申明(访问视为同意此申明)

1.在网站平台的任何操作视为已阅读和同意网站底部的版权及免责申明
2.部分网络用户分享TXT文件内容为网盘地址有可能会失效(此类多为视频教程,如发生失效情况【联系客服】自助退回)
3.请多看看评论和内容介绍大数据情况下资源并不能保证每一条都是完美的资源
4.是否访问均为用户自主行为,本站只提供搜索服务不提供技术支持,感谢您的支持
意见反馈 联系客服 返回顶部

登录注册找回密码

捐赠账单

可选择微信或支付宝捐赠

*请依据自身情况量力选择捐赠类型并点击“确认”按钮

*依据中国相关法规,捐赠金额平台将不予提供发票

*感谢您的捐赠,我们竭诚为您提供更好的搜索服务

*本着平台非营利,请自主选择捐赠或分享资源获得积分

*您的捐赠仅代表平台的搜索服务费,如有疑问请通过联系客服反馈

*推荐用chrome浏览器访问本站,禁用360/Edge浏览器

*请务必认真阅读上诉声明,捐赠视为理解同意上诉声明

账号剩余积分: 0
啥都没有哦