数据集
脑肿瘤区域检测,我们直接采用kaggle公开数据集,Br35H
数据中已对医学图像中脑肿瘤位置进行标注
数据集我已经按照YOLO格式配置好,数据内容如下
数据集中共包含700张图像,其中训练集500张,验证集200张
模型训练
采用YOLOv9模型进行训练
之后修改数据集配置文件: 需要修改数据集路径以及各个类别对应的名称
之后修改训练代码中的配置:
主要修改数据集路径与预训练模型路径,我用的是yolov9-c.pt
训练效果
这些都是模型训练产生的结果,被放在了runs\train\exp 文件下
图像用户界面
网上可以可到很多开源界面,我这里提供了一个,放在了文件夹中
完整项目链接放在了评论区
项目链接