YOLOv7-pose
神笔馬良
正在进化的亚古兽
展开
-
YOLOv7固定预测框的颜色。
YOLOv7固定预测框的颜色。原创 2023-07-31 16:39:41 · 365 阅读 · 0 评论 -
YOLOv7预测框的粗细怎么调整
在detect.py里面加入这一行也是没有用的(这个方法是在其他博主那看到的);修改其他行处的 line_thickness也是没有用的;最后试了一下将thickness=tl修改为thickness=10是有用的。修改的代码处于的位置。yolov7/utils/plots.py中的def plot_one_box(x, img, color=None, label=None, line_thickness=10)函数里面。解决办法:修改cv2.rectangle()函数中的参数值。原创 2023-07-07 11:49:16 · 789 阅读 · 0 评论 -
Labels和Images的文件不对应/如何从文件夹2中找出与文件夹1中对应的文件。
其实:我上面的工作对于训练YOLOv7-pose代码都是多余的,我漏看了原来下载的coco2017labels-keypoints数据标签已经给出来了一个txt文件,如图6所示。我懒得找代码了(我代码功底不太好,一般代码都是网上找的),我把全部内容复制到了word中,然后使用全部替换功能,将.txt后缀全部删掉。替换后的得到jpg_name.txt,替换后的结果如图5所示。解决办法:第一步,我读取出图1标签中文件中 所有的txt文件的名字,并将其,写入到了一个名字为txt_name.txt文件中。原创 2023-05-24 23:43:21 · 223 阅读 · 0 评论 -
深度学习算法中的数据摆放格式
问题描述:在跑关键点检测的算法时,发现一直报错,然后改变了一下数据的摆放格式,然后代码就跑通了。可能我们再跑代码时,需要按照官方数据的摆放方式。总结了几个常用算法的数据摆放格式。第一个是YOLOv7-pose,主要是用来检测人体姿态关键点的。第二个是YOLOv8-pose,也是用来检测人体关键点的。第三个是YOLOv5,用来检测目标检测。原创 2023-05-24 20:24:50 · 49 阅读 · 0 评论 -
YOLOv7-pose运行过程中出现RuntimeError: CUDA out of memory. Tried to allocate 8.14 GiB的错误。
问题描述:在跑YOLOv7-pose代码时,出现了下面这个错误RuntimeError: CUDA out of memory. Tried to allocate 8.14 GiB (GPU 0;事实上,之前跑YOLOv5代码的时候也出现了这个报错,具体YOLOv5的哪个版本就忘了。怀疑是参数量大的版本,因为YOLOv7-pose的权重文件是161.2MB。之前遇到这个错误,网上看见的说法都是说电脑显存不够用,估计换个更高配置的显卡也能解决这个问题。解决办法:我尝试把电脑重启了一下,然后问题就解决了。原创 2023-05-24 15:25:48 · 511 阅读 · 0 评论