- 博客(20)
- 收藏
- 关注

原创 最新版YOLOv6训练自己的数据集(超详细完整版!)
接着上篇文章继续写,本篇文章讲如何训练自己的数据集。从官网下载YOLOv6源码:meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial applications. (github.com)https://github.com/meituan/YOLOv6权重根据自己需求下载,我下的是yolov6s.pt将tools文件夹中的train.py放主目录下,再创一个myself.yaml文件
2022-06-29 16:56:40
25783
122

原创 手把手教你运行YOLOv6(超详细)
YOLOv6 是美团视觉智能部研发的一款目标检测框架,致力于工业应用。本框架同时专注于检测的精度和推理效率,在工业界常用的尺寸模型中:YOLOv6-nano 在 COCO 上精度可达 35.0% AP,在 T4 上推理速度可达 1242 FPS;YOLOv6-s 在 COCO 上精度可达 43.1% AP,在 T4 上推理速度可达 520 FPS。在部署方面,YOLOv6 支持 GPU(TensorRT)、CPU(OPENVINO)、ARM(MNN、TNN、NCNN)等不同平台的部署,极大地简化工程部署时的
2022-06-28 10:52:26
33454
54
原创 Haclon 实现模版匹配
首先拍一张图片作为模版,然后对需要检测的图片进行目标匹配,可区分正反,并输出正确目标的中心点位置,以及输出目标的旋转角度。
2024-11-18 10:09:36
202
1
原创 Openvino2024.3版部署YOLO V5 (C++)
到这一步openvino的环境算是全部装好了,前提是上面的安装都没有报错,验证的话先使用pycharm进行转换,pt->onnx->xml/bin。cmd,进入虚拟环境->D盘openvino_2024文件夹中,运行setupvars.bat,再把下面的都运行,可能需要先更新一下你的pip。下载好后解压出来,放C盘D盘都一样,我放在D盘了,这个文件夹名字有点长,自己修改成openvino_2024。运行效果图就不展示了,代码没有问题的,如果报错了,肯定是你自己的细节问题,不懂的下面评论。
2024-08-22 10:57:57
2177
23
原创 TensorRT框架解析
简介:NVIDIA TensorRT 的核心是一个C++库。这有助于在 NVIDIA 图形处理单元 (GPU) 上进行高性能推理。
2022-12-16 10:09:07
897
原创 YOLOv7 训练报错 IndexError: list index out of range
解决YOLOv7中的报错:IndexError : list index out of range
2022-07-18 13:53:29
12343
15
原创 YOLOv6训练时报错解决方法
解决方法: 是因为你yaml文件中数据集的路径不对,YOLOv6读取的数据集格式与YOLOv5不同将数据集放在主目录中,mydata是我的数据集 yaml文件
2022-07-01 09:47:16
3366
15
原创 YOLOv5更换激活函数(FReLU+SE机制)
本次我们在卷积层添加SE注意力模块来改进YOLOv5算法,同时用FReLU替换SILU激活函数,并通过添加跳转连接来改进PANet特征融合网络。
2022-06-01 13:34:57
10101
28
原创 TensorRT生成engine文件异常([TRT] Network::addScale::434, condition: shift.count > 0 ?)
Loading weights: ./best.wts[05/10/2022-08:41:14] [E] [TRT] Parameter check failed at: Network.cpp::nvinfer1::Network::addScale::434, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)这个错误是wts转engine文件时的报错,在网上找了好久也没有找到.
2022-05-10 09:12:05
5286
18
原创 基于YOLOv5的目标检测——C++版本(展示瑕疵数量、类别,坐标,置信度)
功能:读取文件夹中的所有图片,对图片进行检测后,保存在另一个文件夹中先上效果图上面先输出检测图片需要的时间下面的4代表瑕疵数量,第一列是瑕疵ID,后面是瑕疵类别、坐标、置信度代码主函数 char class_names[55] = "./defects.names"; char weights[55] = "./best.torchscript.pt"; //char source[55] = "./A-M255.bmp"; string source = "
2022-04-19 11:04:28
4391
5
原创 Windows YOLOv5-TensorRT部署
先去TensorRT官网下载安装包(我的是7.2.1.6版本)TensorRT安装测试1、将TensorRT压缩包解压2、将 TensorRT-7.2.1.6\include中头文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include3、将TensorRT-7.2.1.6\lib中所有lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v..
2022-03-24 17:05:24
6699
17
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人