添加注意力机制:
参考: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)