卷积网路CNN分类的模型一般使用包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型
本次是否感染COVID的检测
代码演示可以看bilibili010期
https://space.bilibili.com/124080712
效果图如下:
整体代码文件如下:
dataset文件夹存放的是是否感染COVID的图像 通过01训练数据集文本生成.py将数据集文件夹下的图像路径保存在当前文件下的test.txt,和train.txt中,再运行02VGG16迁移学习训练模型.py读取txt中的数据进行训练。得到的模型保存为当前文件下的model.ckpt.然后运行03预测.py可以调用上一步保存的模型对单张图片进行识别。
04pyqt界面.py是展示可视化的ui界面,可以通过点击按钮加载图片识别。
欢迎下载