yolov5s-jtbz-npu
声明:资源链接索引至第三方,平台不作任何存储,仅提供信息检索服务,若有版权问题,请https://help.coders100.com提交工单反馈
在KhadasVIM3上使用tengine部署YOLOv5s交通标志识别模型到NPU,平均推理速度为49.83ms。以下是详细的描述:
1. 首先,需要在KhadasVIM3上安装tengine和YOLOv5s模型。可以通过以下命令进行安装:
2. 下载YOLOv5s模型,并将其解压到一个文件夹中。例如,将文件放在`/home/user/yolov5s`目录下。
3. 修改YOLOv5s配置文件,设置输入图像大小、输出类别数量等参数。例如,将输入图像大小设置为`640x640`,输出类别数量设置为`10`。
4. 编写Python脚本,使用tengine引擎执行YOLOv5s模型。在脚本中,需要导入所需的库并调用tengine的API。例如:
5. 运行Python脚本,观察输出结果。根据输出结果,可以对模型进行调整以获得更好的性能。
6. 将训练好的模型上传到KhadasVIM3服务器,并使用tengine引擎执行模型。这样,交通标志识别任务就可以在NPU上高效地进行推理。复现在KhadasVIM3上使用tengine部署yolov5s的交通标志识别模型到NPU上,平均推理速度49.83ms
1. 首先,需要在KhadasVIM3上安装tengine和YOLOv5s模型。可以通过以下命令进行安装:
pip install tengine
pip install opencv-python
pip install yolov5s2. 下载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
访问申明(访问视为同意此申明)
2.部分网络用户分享TXT文件内容为网盘地址有可能会失效(此类多为视频教程,如发生失效情况【联系客服】自助退回)
3.请多看看评论和内容介绍大数据情况下资源并不能保证每一条都是完美的资源
4.是否访问均为用户自主行为,本站只提供搜索服务不提供技术支持,感谢您的支持