自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(18)
  • 收藏
  • 关注

原创 yolov8训练pose(人体姿态)数据集

在ultralytics/models/yolo/pose路径下新建一个pose_train_my.py,将以下内容复制进去。如果你是训练非人体的目标,需要修改关键点的名字,数量,以及目标的类名;运行pose_train_my.py开始训练,参数可以用我设置的,也可以自己修改。标注完毕后的数据集格式,图片和json放一起。数据集的存放位置如图。*** 可以只标注看得见的关节点***看不见的可以不用标注。运行后得到分割好的数据集,以及用于训练的yaml文件。修改数据集路径(*** 需要用绝对路径 ***)

2024-08-07 18:10:15 1631 4

原创 yolov8训练segment(分割)数据集

在ultralytics/models/yolo/segment,这个路径下新建一个segment_train_my.py,复制以下内容。运行后会生成两个文件夹,seg和txt。其中seg是最终的数据集(已经划分好的)。里面有一个yam文件,用于训练时用。训练好的模型保存在ultralytics/models/yolo/segment/runs,这个路径下。在项目根目录下新建一个segment_json2txt.py,复制以下内容。注意修改路径和参数,修改完运行,开始训练。

2024-08-02 16:11:20 1089

原创 yolov8训练detect(目标检测)数据集

注意,参数设置优先是results = model.train(task='detect',data="datasets/detect/dataset.yaml", model='yolov8n.pt', epochs=300, imgsz=640, batch=16, workers=0, resume=False)这个里面的。根据自己情况修改参数,或直接去ultralytics/cfg/default.yaml路径下修改,是 一样的。至此,数据集转换完成。

2024-08-01 14:51:13 749

原创 yolov8训练旋转目标检测obb数据集

根据自己的情况修改参数,或者前往ultralytics/cfg/default.yaml,这个路径下修改这个yaml文件里面的参数,都是一样的。3.2 在ultralytics/models/yolo/obb下新建一个obb_train_my.py文件。然后运行data_to_yolo.py,运行完后在labels文件夹下会生成train和val两个文件夹。在项目根目录下创建一个roxml_to_dota.py文件,用于转换成dota能用的obb的txt文件。运行前需修改两个位置!**修改图片后缀**

2024-08-01 11:53:45 1710 2

原创 yolov8官方代码训练模型(coco数据集和voc数据集训练)+onnx模型预测

根据官方的解释,pip 的 ultralytics 库包含了 requirements.txt中的所有库。windows11 和 Ubuntu20.04(建议使用 Linux 系统)首先切换到自己建立的虚拟环境安装 pytorch。参照官网,直接使用以下语句即可导入项目所需要的库。train 和 val 的路径就这样写。2.3 yaml文件。

2023-12-04 16:19:30 15379 27

原创 PadleOCR训练自己的ocr模型之数据标注及导出

输入指令之后,在train_data文件夹下会出现以下文件,其中det是用来训练文字检测的数据集,rec是用来训练文字识别的数据集。此时可以把原来的数据集删去或者移到其他地方。在PaddleOCR根目录下建立train_data文件夹,并且将打标签生成的文件和图片放在该文件夹下。打开终端进入PPOCRLabel的文件夹下,输入以下代码进行数据集的划分。此时文字检测和文字识别的数据集就都制作好了。cd 到 PPOCRLabel 目录下。

2023-11-07 17:13:43 3247 2

原创 YOLOv8-pose之tensorrt版本

1、复制TensorRT-8.4.3.1\bin中内容到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin。4、使用pip install xxx.whl安装TensorRT-8.4.3.1文件夹中框出来的。2、复制TensorRT的include文件夹到CUDA的include文件夹。使用如下脚本:(需要安装ultralytics,直接pip安装即可)这个不影响,只是警告,等待几分钟就转换完成了。(一)模型转换遇到的问题。

2023-11-01 15:11:42 1235 2

原创 paddleocr官方模型转onnx推理预测

链接:https://pan.baidu.com/s/1oA8Cr6LcUJ7tqULRR7ss5Q?链接:https://pan.baidu.com/s/1ktrTCM7pgfWQ6ZSM-6mx_g?那么久修改这个地方,预期多少改多少,比如,以上错误久需要改为48。下载后放到项目目录下,pip install 安装即可。转完后放在一个文件夹里,然后下载这个txt文件放在一起。,直接复制全部代码到一个py文件即可。选择对应自己的python版本。其他的遇到什么安装什么。

2023-10-17 17:06:33 739

原创 onnx模型预测环境安装笔记

CPU版本:直接pip安装GPU版本:先查看自己CUDA版本然后在下面的链接去找对应的onnxruntime的版本查询到对应版本,直接pip安装即可,例如。

2023-09-19 14:35:29 900

原创 paddleocr环境安装以及遇到的问题

CPU版本和GPU版本自己选择链接: 官方安装教程官方说的是推荐使用2.0.1+版本以安装最新版(2023.9.18),最新版为pip install paddleocr==2.7.0.2链接:shapely库选择对应自己的python版本复制下面的文件,直接放入虚拟环境中链接:百度网盘提取码: ufr8然后放入自己的虚拟环境的这个位置注意:如果还是报错,那可能是需要安装 Visual Studio。直接安装即可直接安装

2023-09-18 11:42:20 426

原创 PadleOCR训练自己的ocr模型之训练步骤

一、前期准备1、代码2、标注工具二、环境配置库的安装:在文件配置完后,训练后时看缺什么库就安什么库三、文件配置。

2023-06-26 12:47:27 10949 14

原创 小功能记录单

一、边缘轮廓检测并框出来。

2023-06-15 11:17:43 67 1

原创 yolov5区域检测+计数

具体如何使用、预测、训练等看博主的视频讲解。修改后预测就行了(视频、图片都可以)

2023-02-16 15:00:54 2190 6

原创 文件(夹)的多种处理方法

【代码】文件(夹)的多种处理方法。

2022-12-09 14:26:06 348

原创 Windows下的虚拟机中python程序打包apk

然后把程序复制一下,点击左边栏kivydev目录下找到kivy,在kivy目录下进入accordion。再将"main.py"复制到当前目录下(即**/home/kivydev/kivy/accordion/**目录下)。它会覆盖此目录下的示例——一个main.py文件。因为打包的时候,主程序的名字都是main.py。粘贴我们之前创建的【共享文件夹】位置,将【自动挂载】打勾,点击确定,配置完成。刚安装完的虚拟机界面是这个样子的,是没有任何系统的。找到对应的py文件,放入Windows的共享文件夹就行了。

2022-12-08 11:54:26 3774 6

原创 扩增数据集 图片、xml、json文件

B站的UP主:图像增强软件V2.0,适用于|目标检测|图像分割|图像分类|_哔哩哔哩_bilibili软件下载:Releases · Fafa-DL/Image-Augmentation (github.com)

2022-10-27 16:11:35 805

原创 CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 2.00 GiB total capacity; 960.29 MiB already

input_shape = [320, 320]-----显存不足的话就调小一点(大小为32的倍数)显存不足就调小-----最小为1(有的是最小为2)-------#不能为0#选择是否打开多线程模式 或者 打开几个线程。3、改变num_workers(线程)的数值。2、改变batch_size的大小。1、改变输入图片的大小。错误:训练时显存不足。

2022-09-13 11:15:55 2206

原创 pytorch--GPU环境安装

3、在安装好CUDA和Cudnn之后(详细过程按照置顶的教程走),配置pytorch环境。温馨提示:保证C盘有25G以上的内存,以前准备好系统还原点以防安装错误。0、安装CUDA之前需要安装Visual Studio(2017)1、先看显卡对应的GUDA版本,下载CUDA 版本。2、根据GUDA的版本去找Cudn。4、其他依赖库的安装。视频下有UP主的代码。...

2022-09-01 10:52:33 1096 1

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除