YOLOv8源碼到底在哪?

很多人也想跟修改YOLOv5源碼一樣的方式去修改YOLOv8的源碼，但是在github上面卻發(fā)現(xiàn)找到的YOLOv8項(xiàng)目下面TAG分支是空的，然后就直接從master/main下面把源碼克隆出來(lái)一通修改了，其實(shí)這種方式風(fēng)險(xiǎn)很高，而且也不是正確的選擇，原因在于Master/main分支是工作跟合并的代碼主倉(cāng)，不是發(fā)布的版本代碼，開(kāi)源項(xiàng)目團(tuán)隊(duì)開(kāi)發(fā)者隨時(shí)隨地都在修改代碼，很可能導(dǎo)致引入新BUG，直接使用master/main分支的下面的代碼會(huì)有兩個(gè)致命問(wèn)題：

1. 無(wú)法標(biāo)記版本信息
2. 無(wú)法確信代碼的是否含有致命BUG

C:UsersAdministratorAppDataLocalProgramsPythonPython38Libsite-packages

YOLOv8添加SE注意力模塊

分別修改YOLOv8的模型描述文件yolov8.yaml，模型加載與定義文件task.py與模塊結(jié)構(gòu)化實(shí)現(xiàn)文件block.py，修改的內(nèi)容跟YOLOv5注意力機(jī)制添加完全一致。修改yolov8.yaml文件，這里我直接插入了SENet注意力模塊，修改以后文件與之前的文件對(duì)比如下：

然后修改源碼文件task.py與block.py 實(shí)現(xiàn)SENet注意機(jī)制解析支持。全部完成之后就可以直接開(kāi)始訓(xùn)練模型了，這里基于我的DM碼數(shù)據(jù)集完成模型結(jié)構(gòu)修改之后的YOLOv8注意力模型的訓(xùn)練，命令行跟以前唯一不同的是需要從指定修改的YAML文件開(kāi)始訓(xùn)練，才是對(duì)的，訓(xùn)練的命令行如下：

yolo train model=yolov8.yaml data=dm_dataset.yaml epochs=15 imgsz=640 batch=1

ONNX導(dǎo)出與推理測(cè)試

然后導(dǎo)出ONNX格式模型，設(shè)置一下opset=11 會(huì)有好處。

yolo export model=runsdetect	rain10weightsest.pt format=onnx

導(dǎo)出ONNX格式也可以啦。

預(yù)測(cè)推理

yolo predict mode=detect model=runsdetect	rain10weightsest.onnx

從此又解鎖了YOLOv8模型結(jié)構(gòu)+各種注意力機(jī)制的即插即用從訓(xùn)練到導(dǎo)出部署技能。





審核編輯：劉清