yolov8添加注意力机制并使用预训练加载模型训练

添加注意力机制:

参考:https://github.com/shuai-cao/YOLOv8_cs
https://blog.csdn.net/qq_44224801/article/details/130059070?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522171116498416800184159769%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=171116498416800184159769&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_ecpm_v1~hot_rank-3-130059070-null-null.142v99pc_search_result_base7&utm_term=yolov%E6%B7%BB%E5%8A%A0%E6%B3%A8%E6%84%8F%E5%8A%9B%E6%9C%BA%E5%88%B6&spm=1018.2226.3001.4187

print yaml加进去了,但是训练的时候打印模型就没有,我的解决方法

先按照这个人的:
https://blog.csdn.net/qq_42589613/article/details/135573196?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522171124142216800226512562%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=171124142216800226512562&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_ecpm_v1~rank_v31_ecpm-9-135573196-null-null.142v99pc_search_result_base7&utm_term=yolo%E6%B7%BB%E5%8A%A0%E6%B3%A8%E6%84%8F%E5%8A%9B%E6%9C%BA%E5%88%B6%E5%90%8E%E9%A2%84%E8%AE%AD%E7%BB%83%E6%A8%A1%E5%9E%8B&spm=1018.2226.3001.4187

训练模型的这个:
model = YOLO(“yolov8_cbam.yaml”) # build a YOLOv8n model from scratch
model = YOLO(“/home/featurize/work/ultralytics/yolov8n.pt”)
results = model.train(data=‘string.yaml’, epochs=200)

改成:
from ultralytics import YOLO
model = YOLO(“yolov8_cbam.yaml”).load(“/home/featurize/work/ultralytics/yolov8n.pt”)
results = model.train(data=‘string.yaml’, epochs=200)

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值