104基于深度学习识别是AI生成还是真实图片

本期给大家介绍的是基于深度学习识别是AI生成还是真实图片，效果图如下：

这是运行python 03pyqt.py的可视化界面，通过点击左侧可以加载要识别的检测对象图片，点击中间的识别按钮，会识别出结果，识别结果在右侧显示，当前识别的图片是真实的图片，非假图片。

代码下载和视频演示地址：

104基于深度学习识别是AI生成还是真实图片_哔哩哔哩_bilibili

代码整体是非常简便的，总共三个py部分和一个数据集在data文件夹下。

data文件夹里面一共是2个文件夹分别是真图和AI生成的图各5w张。有兴趣的小伙伴也可以自行扩充数据集，或训练自己的数据集。

运行python 01数据集文本生成制作.py 会在logs文件夹下生成2个txt文本，分别存放了图片路径和对应的标签。然后运行python 02train.py就会训练这个txt文本里面的数据，并将训练好的模型也保存在本地。

最后模型也是保存在logs文件夹下。

后运行python 03pyqt.py即可加载训练好的模型，真假图片进行识别。