项目简介
本项目是基于 PyTorch 和 QT 开发的胸部 CT 与 X - ray 辅助阅片系统。借助深度学习技术对胸部 CT 和 X - ray 影像进行分类预测,辅助医生开展阅片工作。尽管系统存在图像通道差异可能导致闪退、容错性较低等问题,但仍可为相关领域的学习和研究提供参考。
项目的主要特性和功能
- 影像分类预测:支持胸部 CT 的三分类预测以及 X - ray 的二分类预测。
- 模型性能展示:提供验证集的混淆矩阵,直观呈现模型的分类效果。
- 特征可视化:可绘制热力图展示影像的特征分布。
安装使用步骤
准备工作
- 确保已下载本项目的源码文件。
- 从 Google 硬盘下载可执行程序文件与模型权重:点击下载。
- 下载 ResNet50 预训练模型权重:点击下载。
- X - ray 数据集可从 此处 下载,CT 数据集可从 此处 下载。
目录结构搭建
在 System
文件作为根目录下,新建两个目录 ./NET50
和 ./2CData
,将从 Google 硬盘下载的模型权重放置在 ./NET50
目录下,系统运行时会使用 ./2CData
目录。最终目录结构如下:
System
| class2_indices.json
| class3_indices.json
| main.exe
| resnet50 - pre.pth
| 图标.ico
|
+---2CData
\---NET50
ResNet50_32_2.pth
ResNet50_64_3.pth
运行系统
准备就绪后,直接运行 main.exe
即可启动系统。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】