- 博客(18)
- 资源 (1)
- 收藏
- 关注
原创 YOLOv8 修改 IoU(CIOU, DIOU, SIOU, EIOU, Focal_EIOU) YOLOv8 obb修改CIOU
修改上述两个地方的IOU是不能用于旋转框的,用于旋转框的IOU,目前这个版本的yolo v8只有CIOU可用,同样,在ultralytics/utils/loss.py文件中我们找到RotatedBboxLoss(BboxLoss)这个类,大概在118行处有一行这样的代码。需要注意的是,现阶段yolo v8的代码几乎每天都在变动,现在日期是24年4月14日,因此我的代码改动也是基于这几天最新的yolo v8版本的。只要这里面有的IOU,你都可以在上面的两个地方进行修改使用。我们将其CIOU=Ture。
2024-04-14 12:54:08
2565
5
原创 yolo测试集验证&模型转换
模型训练完成之后将pt文件和yaml文件的地址复制到下面的代码中,在命令行中运行。此为在测试集上的模型准确度,一般会比训练集上面的低。下面的代码用于将训练好的pt文件转换为onnx格式。test 文件夹里面包含已经标注好的数据。
2024-03-12 21:34:43
419
1
原创 yolov8添加CBAM注意力机制
解决方法:复制yolov8的task.py到conda\envs\yolo_track\Lib\site-packages\ultralytics\nn,yolo_track为你的环境名字,把这里面的task.py覆盖掉即可。第六步:在myolov8_Att_CBAM.yaml里面添加注意力机制,添加的位置不同,效果也不一样,这个需要自研究一下,我添加在了第五层,在head这个部分标红的地方进行了修改,因为我们加了一次,所以这些也需要加一层,:这是输入的索引,表示该层的输入来自于模型的上一层。
2024-01-12 12:44:39
4048
7
原创 yolo模型效果评估
如图所示,训练后能够在终端看到你最好的权重在哪个目录使用yolov8进行训练一般会产生如下文件:其中,weights文件夹内会存放一个最好的权重文件和最后一次的权重文件 args.yaml: 保存此次训练的相关参数 confusion_matrix.png: 混淆矩阵是对分类问题预测结果的总结,它通常包括以下内容: TP(True Positive): 将正类预测为正类数 即正确预测,真实为0,预测也为0 TN (Tru
2023-12-20 17:33:05
710
1
原创 虚拟环境下的安装包都来自于基本环境
此时,需要修改我们新创的环境包的路径,在自己的环境Lib文件下找到site.py文件,我的新环境是D盘,查看之后发现包位置是C盘我的本地环境。同样的改为你新环境下的路径即可。
2023-12-07 22:31:48
211
1
原创 使用yoloV5训练自己的模型
-cache-images:是否提前缓存图片到内存,以加快训练速度,默认False。--single-cls:数据集是否只有一个类别,默认False。--multi-scale:是否进行多尺度训练,默认False。--sync-bn:是否使用跨卡同步BN,在DDP模式使用。--image-weights:使用加权图像选择进行训练。--evolve:是否进行超参数进化,默认False。--exist-ok:模型目录是否存在,不存在就创建。--rect:是否采用矩形训练,默认False。
2023-11-26 16:22:05
258
1
原创 使用conda创建不同的虚拟环境
第二步:输入conda create -n pytorch python=2.8 (pytorch是环境名称可以随便取)第一步:打开我们的终端。
2023-09-25 10:16:08
560
1
原创 selenium的安装和初步使用
在使用selenium之前,驱动是不可或缺的,首先查看自己谷歌浏览器的版本:我们选择和自己版本号相同的进行下载即可,需要注意的是前面的大版本号对住就行,最后几位对不住也没有关系。我的是windows系统,然后点击下载下载完之后,解压,会有一个chromedriver.exe,这个就是我们需要使用的驱动。
2023-04-06 18:55:30
2032
原创 python爬虫遇到验证页面
在练习爬虫的时候爬取一个网页的源码遇到了一个这样的问题!一开始搞了好久,想着总不能在写个自动拖验证码的脚步吧,这我也不会啊,本来我以为是我的ua伪装的不好。浏览器发现了,于是我写了个随机ua的headsimport urllib.request,urllib.errorfrom fake_useragent import UserAgentdef askURL(URL): heads = { "User-Agent": str(UserAgent().random
2022-01-14 13:39:21
991
原创 java开发IDEA的使用和下载
第一步:下载 链接在这里:Download IntelliJ IDEA: The Capable & Ergonomic Java IDE by JetBrainsDownload the latest version of IntelliJ IDEA for Windows, macOS or Linux.https://www.jetbrains.com/idea/download/#section=windows直接下载安装就行,和大部分软件一样,记得改默认安装路径。(实在不会...
2022-01-08 15:50:17
411
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅