基于ResNet和YOLOv8的病理性高度近视检测
在最后的webui中,使用者可以输入患者名称并上传患者的超高清眼底图彩照
模型会首先进行分类(是否有豹纹状改变)然后会用YOLOv8来识别病灶
识别完成后,会总结所有的结果并同时输出到webui和pdf文件中
数据集分为两部分:
分类算法会对患者的眼底图进行分类:有无豹纹状改变
在dataset/classification文件夹下有两个子文件夹:
├─dataset
├─classification
├─train <- 训练集
│ ├─no <- 没有豹纹状改变
│ └─yes <- 有豹纹状改变
└─val <- 测试集
├─no
└─yes
英文名 | 中文名 | 标签 |
---|---|---|
peripapillary atrophy | 视盘萎缩斑 | PPA |
macular degeneration | 黄斑萎缩 | MD |
vitreous opacities | 玻璃体浑浊 | weiss |
drusen | 玻璃膜疣 | DR |
optic disc | 正常视盘 | OD |
fuchs dystrophy | 福斯氏角膜内皮营养不良 | Fuchs |
高度近视患者的症状表现,高度近视眼因眼轴的过度伸长,可引起眼底的退行性改变。 黄斑部可发生形成不规则的、单独或融合的白色萎缩斑,有时可见出血。
AMD (MD) 又叫年龄相关性黄斑变性,能够影响视网膜中心区域黄斑。AMD在绝大多数情况下见于双眼,病程为进行性,最终可导致中央视觉永久丧失,也是工业化国家中50岁以上人群视力低下的首要原因。
玻璃体浑浊又称飞蚊症。 先天残留于玻璃体内的胚胎细胞或组织、视网膜或葡萄膜的出血侵入玻璃体内,高血压、糖尿病、葡萄膜炎的出血或渗出物侵入玻璃体内,老年人高度近视眼的玻璃体变性,均可导致玻璃体液化而浑浊。 其他如眼外伤、眼内异物存久留、寄生虫或肿瘤等也可发生玻璃体浑浊。
玻璃膜疣是胶样或透明的小体,是一种发生在脉络膜视网膜的一种变性疾病,是色素上皮细胞异常代谢产物在视网膜上的异常沉积所致。
正常视盘略呈椭圆形、淡红色、边界清楚,中央呈漏斗形凹陷,色泽稍淡,称为生理凹陷。
在福斯氏角膜内皮营养不良患者中,液体积累在眼睛前方的透明层(角膜)中,造成角膜肿胀和增厚。 这可能导致眩光、视力模糊或浑浊以及眼睛不适。 福斯氏角膜内皮营养不良通常侵袭双眼,可能在多年中导致视力逐渐下降。
pip install -r requirements.txt
python classification.py
或者使用jupyter notebook
使用ResNet18加预训练模型结果如下:
python yolo.py
或者使用jupyter notebook
使用yolov8s预训练模型混淆矩阵效果如下:
Usage: python webui.py [options]
A common command: python webui.py
-l --language Language of the UI site