- 博客(6)
- 收藏
- 关注
RSS订阅原创 yolov7训练train_aux.py报错：RuntimeError: indices should be either on cpu or on the same device as the in
把loss.py文件中的1387行“matching_matrix=torch.zeros_like（cost）”更改为matching_matrix=torch.zeros_link（cost，device=“cpu”）把loss.py文件中的1541行"matching_matrix = torch.zeros_like(cost) 更改为。
2023-11-14 15:09:09
232
原创 NMS(非极大值抑制)python实现
基于深度学习的目标检测算法都会从一张图像中得到属于某个类别的多个矩形框，并且会得到每个矩形框属于这个类别的概率。在检测结果中，往往会出现很多矩形框检测到的都是同一个目标的情况，但其实每个目标只需要一个最准确的矩形框 就可以了。每个物体只保留最准确（置信度最高）的一个矩形框，其余的全删除。所谓非极大值抑制：依靠分类器得到多个候选框，以及关于候选框中属于类别的概率值，根据分类器得到的类别分类概率做排序，具体算法流程如下：（1）将所有框的得分排序，选中最高分及其对应的框。
2023-08-27 10:34:48
137
原创 yolov5训练没有发现数据集的问题
我们在使用yolov5训练数据集的时候，如果数据集是VOC或是coco分布和命名格式的不需要修改，而其他格式的数据集需要在datasets.py文件中修改。在datasets.py文件中找到函数："def img2label_paths(img_paths):"将函数里面的images和labels换成自己数据集里面图片和标签所在文件夹的名字即可。我的是yolov5-5.0，在341行。
2023-07-04 10:53:29
534
原创 yolov5训练报错“data = self.stream.read(size)UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0x80 in “
网上查了一下说是无法读取yaml文件，可是我数据集里面的标签全是txt格式的，不明白为什么会出现这种情况？只需要把train.py第63行代码的"with open(opt.data) as f:"在使用yolov5-5.0训练VisDrone2019数据集的时候，出现了下面的报错。换成"with open(opt.data,'rb') as f:" 就可以了。后来网上查了一下好在找到了解决办法(虽然不知道为什么？
2023-07-04 10:39:53
617
原创 Ubuntu22.04一键安装ROS2
wget http://fishros.com/install -O fishros && . fishros
2023-06-19 17:38:05
121
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人