1、立项背景和目标
乳腺癌病理人工诊断效率低、易出现漏诊误诊,基层医师阅片水平参差不齐,同时传统深度学习模型判定过程不透明,难以用于临床参考。本项目目标是搭建 CNN-RNN 混合识别模型,开发配套可视化桌面软件,精准完成乳腺病理图像良恶性判定,通过热力图直观标注病灶区域,打造轻量化 AI 辅助诊断工具,提升病理筛查效率,为医护人员提供可解释的智能判断依据。
2、软件功能、核心功能模块的介绍
本软件基于 Python Tkinter 开发,包含四大核心模块。一是图像上传预处理模块,支持本地病理图片读取、尺寸归一化、像素标准化;二是 AI 模型推理模块,加载训练完成的 CNN-RNN 模型,快速完成良恶性分类并输出置信概率;三是 Grad-CAM 热力可视化模块,反向计算特征权重,生成病灶高亮热力图;四是交互展示模块,分区呈现原图、热力图、预测结果与运行进度,界面简洁易上手,无需编程基础即可操作。
3、业务流程、功能路径描述
完整使用路径:打开软件主界面→点击上传按钮选中本地病理图像,系统自动完成图片预处理→后台调用 AI 模型进行特征提取与推理运算,同步计算热力图层→界面左侧展示原始病理图,右侧实时加载病灶热力图,底部显示良性 / 恶性判定结果、预测可信度数值与执行进度。使用者可查看 AI 识别结果与病灶定位,核验判断逻辑,完成单次辅助诊断,可重复上传多张图片进行批量测试使用。
1、整体架构、设计思路与技术栈
整体分为界面交互、图像预处理、模型推理、热力可视化四层架构。
界面使用Tkinter,图像处理用 OpenCV、PIL;算法基于TensorFlow搭建 CNN-RNN 网络,依靠 GRU 完成时序特征提取,用 Grad-CAM 实现结果可视化,依托 RTX 显卡完成模型 GPU 训练。
2、本人负责模块与量化结果
我独立完成模型搭建、界面开发与整体联调。最终 CNN-RNN 模型测试准确率 94.6%,AUC=0.99;软件单张图片推理耗时 0.3 秒,可打包 exe 直接运行,对比基础 CNN 模型准确率提升 7.9%。
3、遇到的难点及解决办法
CNN 与 RNN 维度不匹配:新增 Lambda 层重塑特征维度,完成网络拼接;
模型训练过拟合:搭配数据增强、Dropout、早停策略,稳定模型泛化能力;
推理时界面卡死:采用多线程分离界面与运算线程,解决程序卡顿问题;
热力图定位不准:优化梯度计算逻辑,精准锁定病灶高亮区域。