- 博客(4)
- 收藏
- 关注
原创 YOLOv8添加DCNv3可变形卷积报错张量不匹配
get_reference_points 与 _generate_dilation_grids中的 dtype=torch.float32 替换为 dtype=torch.float16,让代码在进行dcnv3相关的数据计算时的数据类型为16位浮点型而不是32位浮点型,让它在gpu上计算而不是cpu上计算,这样就能让 ref 与 grid 参数在gpu上进行计算。导致当我们把数据放在gpu上训练时要求的数据是16位,而dcnv3内的数据是32位,会放在cpu上进行训练,结果会导致一个错误是数据。
2023-10-10 20:21:03 1470 6
原创 YOLOV8数据集运行:用coco128数据集格式跑VOC数据集(V8运行各类数据集万能模板)
以下代码将voc数据集变成yolo格式,将voc数据集的图片存放到images文件中,并且按比例划分为训练集,验证集和测试集,比例为8:1:1。而将voc数据集标注文件原本的xml格式转换为txt格式并存放在labels文件夹中,里面的标签文件对于images文件夹中的训练集,验证集和测试集的标签文件;注意:需在文件夹中提前准备好images文件夹和labels文件夹,并在其中创建train,val,test文件夹。如果不需要test文件,将代码中test删除,把对应的比例划分给train或者val。
2023-09-20 16:25:02 1518 2
原创 YOLOV8报错:no labels
已经将images修改为JPEGImages但还是报错no labels。按照要求也生成了train.txt,val.txt,test.txt;而将voc标签的xml格式转换为了txt格式。
2023-09-19 22:43:34 633 3
原创 虚拟环境下numpy不存在问题
用pip安装了numpy,但是在解释器为虚拟环境时运行确报错:不存在numpy。ModuleNotFoundError: No module named 'numpy'这是pip的安装路径问题,每一个解释器都相当于一个容器,只有在虚拟环境这个容器中有numpy的时候运行才不会报错:不存在numpy,但是我们用pip下载的库默认都是在anaconda/lib/site-packages下,而不是虚拟环境envs/虚拟环境名/lib/site-packages下,所以想要在虚拟环境下使用numpy,必须
2022-10-27 11:22:33 2353 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人