- 博客(3)
- 收藏
- 关注
原创 向Yolov8添加注意力机制报错:“Keyerror”的解决办法
在我们向Yolov8中添加注意力机制时,即便我们正确的在nn/modules/init.py以及nn/tasks.py中正确的导入了注意力机制,仍然会报错:“Keyerror”,似乎仍然是注意力机制仍然没有正确的导入。实际上这是因为我们在配置虚拟环境时,使用了代码在这个过程中,“ultralytics”包会被安装到虚拟环境中。在这里我们一定要注意区分,虚拟环境中的“ultralytics”以及我们克隆的代码的“ultralytics”。
2024-04-15 21:38:24
1824
7
原创 在Yolov8中使用自己的网络结构以及预训练权重进行训练的方法
来进行训练,并且在default.yaml文件内设置网络结构、预训练权重、训练轮次等内容。当我们直接在从Github上获取的预训练权重基础之上进行训练且不需要更改网络结构时,直接将model指向权重的.pt文件即可。但是我们时常需要使用自己优化后的网络结构配合从Github上获取的预训练权重来进行训练。对此我们只需要对default.yaml文件进行以下修改即可。除了使用train.py以及命令行进行训练以外,我们还可以使用。
2024-04-14 00:34:18
2010
5
原创 在使用yolov8训练时调取tensorboard报错:TypeError: MessageToJson() got an unexpected keyword argument including
在使用yolov8训练时调取tensorboard报错:TypeError: MessageToJson() got an unexpected keyword argument including。这是由于tensorboard版本过高导致,只需要将tensorboard版本降低即可。博主降低至2.12.0便可以正常使用。降低tensorboard版本为2.12.0。
2024-03-25 21:49:21
1337
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人