本次是否感染COVID的检测
代码下载和视频演示地址:010基于vgg的CT_COVID与CT_NonCOVID二分类识别_哔哩哔哩_bilibili
效果图如下:
整体代码文件如下:
dataset文件夹存放的是是否感染COVID的图像 通过01训练数据集文本生成.py将数据集文件夹下的图像路径保存在当前文件下的test.txt,和train.txt中,再运行02VGG16迁移学习训练模型.py读取txt中的数据进行训练。得到的模型保存为当前文件下的model.ckpt.然后运行03预测.py可以调用上一步保存的模型对单张图片进行识别。
04pyqt界面.py是展示可视化的ui界面,可以通过点击按钮加载图片识别。
注:
下载本代码环境自行安装
如需远程安装环境运行,
或逐行代码注释(小白也能快速掌握理解代码),
或其他需求
可定制。