【DL】YOLO模型改進記錄

SaTsuki26681534發表於2024-12-03

#1 瞭解YOLOv8的專案結構

改進的時候主要從三個地方中改進

cfg/models/v8/yolov8.yaml

yaml檔案用於定義模型每個部分都由哪些部件組成,以及這些部件的引數值

nn/moudles

nn/models中用於改進元件內部結構

nn/tasks.py

用於定義各種任務的前向傳播過程,可以說定義了模型的結構

相關文章