數據集
腦腫瘤區域檢測,我們直接采用kaggle公開數據集,Br35H
數據中已對醫學圖像中腦腫瘤位置進行標注
數據集我已經按照YOLO格式配置好,數據內容如下
數據集中共包含700張圖像,其中訓練集500張,驗證集200張
模型訓練
采用YOLOv9模型進行訓練
之后修改數據集配置文件: 需要修改數據集路徑以及各個類別對應的名稱
之后修改訓練代碼中的配置:
主要修改數據集路徑與預訓練模型路徑,我用的是yolov9-c.pt
訓練效果
這些都是模型訓練產生的結果,被放在了runs\train\exp 文件下
圖像用戶界面
網上可以可到很多開源界面,我這里提供了一個,放在了文件夾中
完整項目鏈接放在了評論區
項目鏈接