yolov5+opencv+java:通过DJL在maven项目中使用yolov5的小demo

目录

前言

这篇博客主要是介绍如何通过djl在java中调用yolov5进行推理,顺便也学习了一下在java上的opencv api。
Deep Java Library是由亚马逊(Amazon)提供的一个深度学习工具包,能够让java开发者在java上调用目前主流的深度学习框架,像pytorch、tensorflow、mxnet、paddlepaddle(飞桨居然也有份??),也包括onnx格式的模型。

环境

导出yolov5s模型

这次demo就直接使用yolov5s的预训练模型。yolov5项目本身就自带了非常完善的模型导出脚本,yolov5的5.0发行版也比之前的版本完善很多。
yolov5的模型导出脚本是models/export.py文件,
在这里插入图片描述
导出之前需要设置一下

  • 权重文件的位置
  • 输入图片的尺寸
  • 是否要输出bbox
  • 模型所在设备
    在这里插入图片描述
    上图红色的框按我的进行设置就行了,绿色的框根据自己的情况进行设置。
    设置好以后运行代码就可以在和权重文件相同的位置找到生成的torchscript模型权重。
    在这里插入图片描述

编写Maven项目

  • 2
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
YOLOv5是一种基于深度学习的目标检测算法,可以快速准确地检测图像的不同目标。OpenCV是一个流行的计算机视觉库,提供了丰富的图像处理和分析功能。Java是一种常用的编程语言,可以用于开发各种应用程序。 如果想在Java使用YOLOv5来进行印章识别,首先需要安装配置OpenCVYOLOv5的相关环境。可以使用JavaCV库来方便地在Java调用OpenCV的功能,同时也可以使用TensorFlow Java API来加载和使用YOLOv5的模型。 准备工作完成后,我们可以通过以下步骤进行印章识别: 1. 加载和初始化YOLOv5的模型。这包括加载模型文件、配置文件和权重文件,并根据需要进行参数设置。 2. 使用OpenCV读取待识别的图像,并进行预处理。预处理可以包括图像的缩放、归一化等操作,以便符合模型的输入要求。 3. 将预处理后的图像输入到YOLOv5模型进行目标检测。模型将输出图像检测到的印章的位置、类别和置信度等信息。 4. 根据模型的输出结果,可以根据置信度进行筛选和过滤,以得到最可信的印章检测结果。 5. 可以使用OpenCV的绘图功能,在原始图像上标注出检测到的印章位置和类别等信息,以便进行可视化展示或进一步的后续处理。 通过以上步骤,我们就可以实现在Java环境下使用YOLOv5OpenCV进行印章识别。当然,具体实现的细节还需要根据实际情况进行调整和完善,比如模型的训练和优化等。此外,还需要考虑算法的性能和效果等因素,以便得到更好的印章识别结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值