本次是否感染COVID的检测
demo仓库和代码演示可以看bilibili找010期
到此一游7758258的个人空间_哔哩哔哩_bilibili
效果图如下:
![](https://i-blog.csdnimg.cn/blog_migrate/e101c99c351ca9acdc7eede56a1e0c2c.png)
整体代码文件如下:
![](https://i-blog.csdnimg.cn/blog_migrate/1be82d08f7c2c820b4f31b130f49e973.png)
dataset文件夹存放的是是否感染COVID的图像 通过01训练数据集文本生成.py将数据集文件夹下的图像路径保存在当前文件下的test.txt,和train.txt中,再运行02VGG16迁移学习训练模型.py读取txt中的数据进行训练。得到的模型保存为当前文件下的model.ckpt.然后运行03预测.py可以调用上一步保存的模型对单张图片进行识别。
04pyqt界面.py是展示可视化的ui界面,可以通过点击按钮加载图片识别。