- 博客(25)
- 收藏
- 关注
RSS订阅
原创 YOLOv7--复现并训练自己的数据集（简单）
与Yolov7-main同目录下，创建一个datasets文件夹，将ua-detrac数据集上传到该文件夹下。解决方法：找到虚拟环境下的functional.py文件，在下图的位置上加上。2、出现UserWarning: torch.meshgrid 错误。删除数据集labels文件中的.cache文件，需要先删除。注意：这里路径建议使用绝对路径，不然有可能会报错哦！等安装成功之后，就ok了。将yolov7.yaml 中的 nc 改为4。将红框内的改为ua-detrac.yaml。
2024-03-28 13:24:47
1035
原创 ModuleNotFoundError: No module named ‘timm‘
在改进YOLOv8时，报错ModuleNotFoundError: No module named timm。
2023-11-20 12:36:48
369
原创 YOLOv8-训练自己的数据集
model：选择YOLOv8不同的模型配置文件，可选yolov8s.yaml、yolov8m.yaml、 yolov8x.yaml等。mode：选择任务类型，可以选择 训练、验证或者预测的任务，['train' 'val' 'test']task：选择任务类型，可以选择['detect' 'segment' 'classify' 'init']原本红框内是80，则改为4，你的数据集类别个数有多少，就写多少。
2023-11-05 15:21:01
7522
5
原创 YOLOv5改进前后曲线对比图：mAP50，mAP50-90，Loss
其中：result1.csv是原始结果；result2.csv是改进后的结果。
2023-08-31 15:15:16
5370
原创 Yolov5改进之更改损失函数（EIOU、SIOU）
注意：以EIOU为例，当 EIOU=True，alphaIOU>1时，则损失函数是两者的结合，为 a-eiou。当alphaIOU=1时，则损失函数就是EIOU。推荐up主：一个非常棒的B站up主，都去看！大家也可以尝试着添加其他损失函数。
2023-07-06 17:23:21
6989
2
原创 YOLOv5改进之添加Bifpn模块
BiFPN# 两个特征图add操作# 设置可学习参数 nn.Parameter的作用是：将一个不可训练的类型Tensor转换成可以训练的类型parameter# 并且会向宿主模型注册该参数 成为其一部分 即model.parameters()会包含这个parameter# 从而在参数优化的时候可以自动一起优化w = self.w# 三个特征图add操作w = self.w。
2023-06-28 19:35:45
4242
19
原创 解决：RuntimeError: adaptive_max_pool2d_backward_cuda does not have........(添加注意力机制CBAM)
在train.py文件中，大概324行左右，修改下列代码。重新运行 train.py 文件。
2023-06-27 19:52:47
1418
原创 YOLOv5改进之添加CBAM注意力机制
在其后面添加：CBAMBottleneck，C3CBAM。这里是将backbone中的C3全部替换成了C3CBAM。
2023-06-27 16:56:02
3512
2
原创 解决出现CUDA error:out of memory的问题
使用：CUDA_VISIBLE_DEVICES限制一下使用的GPU。CUDA_VISIBLE_DEVICES=0,1 则使用这两张GPU。CUDA_VISIBLE_DEVICES=0 则使用第一张GPU。CUDA_VISIBLE_DEVICES=1，则使用第二张GPU。如何设置CUDA_VISIBLE_DEVICES？例如：有两块GPU，即0,1号GPU，查阅资料，发现是显卡内存不足。
2023-05-24 18:40:51
19818
5
原创 VOC2007数据集转yolo格式
第一步：划分数据集训练集：验证集：测试集=8：1：1新建一个split_data.py文件，用来划分数据集，这里是直接根据图片进行划分的数据集第二步：标签转换：voc-yolo。
2023-05-23 20:27:15
194
原创 yolov5改进---添加YOLOV8中的C2F模块
在yolov5中 modules 文件下，新建一个yolov5s_c2f.yaml 文件。根据上述步骤，即可在yolov5中添加YOLOV8中的C2F模块。在该文件中，将yolov5.yaml中的代码粘贴进来，并将所有的。修改过的 common.py 全部代码如下，可直接复制。8、运行yolov5中的 train.py 文件。7、修改yolov5中的 train.py文件。3、打开yolov5中common.py文件。6、修改yolov5中的yolo.py文件。1、克隆一个全新的yolov5代码。
2023-04-19 12:39:39
3185
4
原创 VOC数据集 转 YOLO数据集
运行上述代码，即可转换成功！如下图所示，直截取了部分。此博客讲解 VOC格式 转 YOLO格式。直接复制下列代码即可转换成功，亲测有效。温馨小提示：记得更改为自己的路径哦！YOLO数据集的格式为 .txt。VOC数据集的格式为 .xml。
2023-04-18 19:22:55
251
原创 上传数据到--云服务器Autodl
我使用了两种方式，第一种为Autodl云服务器中的JupyterLab，第二种方法为使用WinScp工具上传数据。这里着重介绍第二种方式，第一种方式在前面的博客中已经详细介绍了，不在过多赘述。
2023-04-17 16:24:09
3654
1
原创 YOLO-v5训练自己的数据集
例如：打开terminal是，环境显示是base，而你需要进入pytorch环境，则输入以下命令，即可进入pytorch环境中。打开terminal，进入本项目的annoconda环境。运行train.py文件，即可开始训练，训练结果会保存在runs/train文件夹中。这里默认大家已经安装好自己的环境了并配置好了Pycharm。如果环境不是本项目所需的环境，需要激活，进行切换环境。所需要的包安装成功之后，就可以训练自己的数据集了。方法一：可以使用git进行克隆，克隆到本地。进入到了pytorch环境。
2023-04-10 20:44:06
1119
原创 labelImg 使用以及安装过程
labelImg 是一款开源的图像标注的工具，主要用于标注两种数据格式，如下：voc 数据格式：xml文件yolo数据格式，txt文件。
2023-03-22 16:42:25
3079
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人