- 博客(3)
- 收藏
- 关注
RSS订阅原创 向Yolov8添加注意力机制报错：“Keyerror”的解决办法
在我们向Yolov8中添加注意力机制时，即便我们正确的在nn/modules/init.py以及nn/tasks.py中正确的导入了注意力机制，仍然会报错：“Keyerror”，似乎仍然是注意力机制仍然没有正确的导入。实际上这是因为我们在配置虚拟环境时，使用了代码在这个过程中，“ultralytics”包会被安装到虚拟环境中。在这里我们一定要注意区分，虚拟环境中的“ultralytics”以及我们克隆的代码的“ultralytics”。
2024-04-15 21:38:24
2987
11
原创 在Yolov8中使用自己的网络结构以及预训练权重进行训练的方法
来进行训练，并且在default.yaml文件内设置网络结构、预训练权重、训练轮次等内容。当我们直接在从Github上获取的预训练权重基础之上进行训练且不需要更改网络结构时，直接将model指向权重的.pt文件即可。但是我们时常需要使用自己优化后的网络结构配合从Github上获取的预训练权重来进行训练。对此我们只需要对default.yaml文件进行以下修改即可。除了使用train.py以及命令行进行训练以外，我们还可以使用。
2024-04-14 00:34:18
3814
10
原创 在使用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
1578
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人