![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
yolov5实战
一位安分的码农
这个作者很懒,什么都没留下…
展开
-
RuntimeError: Error compiling objects for extension手把手带你解决(超详细)
我在网上找了一些帖子,有让换pytorch版本的,也有其他方法的。我都试过,换pytorch版本确实能起效果,但是这个方法太笨了,我还是推荐使用另外一种方法。1、进入cuda的安装目录。2、找到对应自己pytorch版本的cuda,比如我的有cuda、cuda-11.3、cuda-11.6,但是我的pytorch对应的是cuda(我把cuda11.3命名为cuda了)。.........原创 2022-08-16 10:05:07 · 43916 阅读 · 32 评论 -
配置在关闭本地电脑的情况下远程服务器仍然训练、工作
自己的电脑连上服务器后,训练模型,若这时关闭电脑,则服务器也会随之关闭模型的训练。那么,如何使得自己的电脑关闭的情况下,服务器仍然继续执行训练任务呢?五、需要关闭电脑的时候,使用ctrl+a+d退出screen,再关闭电脑。这个时候不用担心,因为服务器是会继续执行任务的。六、需要查看终端的输出时,重新打开screen即可,执行如下命令。一、用vscode或者cmd的ssh连接到服务器。七、需要删除某个screen时,执行如下命令。二、执行命令,创建screen。三、查看创建的screen。...原创 2022-08-11 10:26:15 · 6616 阅读 · 0 评论 -
Yolov5设置不同类别元素之间使用NMS非极大值抑制
Yolov5默认设置类间元素分别使用NMS,这样一来,会使得图像中的相近区域检测出多个类别目标。如果想要所有的元素一起使用NMS,可以这样设置:进入utils的general.py,在791行添加进行类间NMS时改为True后:...原创 2022-07-11 16:18:13 · 1968 阅读 · 0 评论 -
yolo格式标签在原始图像中可视化显示
有时候,我们想要查看下载的数据集中,别人把锚框标在了图像中的哪个位置。我们可以用此代码,批量查看。效果:原创 2022-07-11 11:27:07 · 3375 阅读 · 8 评论 -
NMS非极大值抑制原理——目标检测
目标检测在使用了基于深度学习的端到端模型后效果斐然。目前,常用的目标检测算法,无论是One-stage的SSD系列算法、YOLO系列算法还是Two-stage的基于RCNN系列的算法,非极大值抑制都是其中必不可少的一个组件。在现有的基于anchor的目标检测算法中,都会产生数量巨大的候选矩形框,这些矩形框有很多是指向同一目标,因此就存在大量冗余的候选矩形框。非极大值抑制算法的目的正在于此,它可以消除多余的框,找到最佳的物体检测位置。非极大值抑制(Non-Maximum Suppression,以下简称NM原创 2022-07-08 17:54:37 · 1381 阅读 · 1 评论 -
Yolov5训练自己的数据集(超详细)
一、从官网下载最新的yolov5代码二、新建VOCData文件夹三、VOCData文件夹结构1、新建Annotations文件夹,存放标签简单的xml文件,应该长这样复杂的xml文件,应该长这个样子2、新建images文件夹,存放图片数据注意:需要观察自己的图片文件的后缀名,后面需要用到,不然可能出现程序找不到图片的情况3、新建split_train_val.py,用来划分训练集与验证集运行之后,会发现VOCData文件夹里面多了ImageSets文件夹里面包括四个txt文原创 2022-07-08 11:06:27 · 10899 阅读 · 11 评论 -
yolov5手动锚定框计算程序
在yolov5目录下新建程序kmean_anchors() 函数中的参数含义:如果在train时,不想程序自动计算锚定框,可以在train.py这样设置:原创 2022-07-07 10:17:57 · 1857 阅读 · 1 评论 -
pytorch实战(一)——环境配置教程(基于Anaconda)
1、下载Anaconda2、创建虚拟环境conda create -n yolov5 python=3.73、conda activate yolov54、换源conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/conda config --add c原创 2022-06-30 15:55:25 · 1219 阅读 · 0 评论 -
yolov5输出检测到的目标坐标信息/位置
找到detect.py,在大概113行,找到plot_one_box # Write results for *xyxy, conf, cls in reversed(det): if save_txt: # Write to file xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(原创 2021-04-01 21:19:31 · 38122 阅读 · 98 评论 -
yolov5报错:RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place
在执行下面代码时python train.py --img 640 --batch 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ''我遇到了此报错Traceback (most recent call last): File "train.py", line 456, in <module> train(hyp, opt, device, tb_writer)原创 2021-03-31 20:28:05 · 16729 阅读 · 32 评论