基于深度学习的猫狗动物识别系统CNNtensorflow卷积神经网络。文章目录1. 环境设置安装依赖2. 数据准备3. 模型训练编写train_model.py4. 创建PyQt5用户界面编写main.py5. 运行项目解释增加视频支持添加视频功能代码及文字仅供参考。为了实现一个基于深度学习的猫狗动物识别系统并使用PyQt5创建用户界面我们可以利用TensorFlow和卷积神经网络CNN来训练模型。以下是详细的步骤和代码示例包括环境配置、数据准备、模型训练以及用户界面设计。1. 环境设置首先确保你的环境中安装了必要的依赖项安装依赖在开发阶段你需要一个Python环境来安装必要的依赖项。创建一个requirements.txt文件记录项目所需的依赖项tensorflow2.6.0 opencv-python-headless4.5.5 PyQt55.15.2 matplotlib3.4.2然后运行以下命令来安装依赖pipinstall-rrequirements.txt2. 数据准备假设你已经有了一个包含猫和狗图片的数据集。如果没有可以下载Kaggle上的猫狗分类数据集。确保数据集按照以下结构组织cat_dog_dataset/ ├── train/ │ ├── cats/ │ └── dogs/ ├── validation/ │ ├── cats/ │ └── dogs/ └── test/ ├── cats/ └── dogs/3. 模型训练我们将使用TensorFlow和Keras API来构建并训练一个简单的卷积神经网络模型。编写train_model.pyimporttensorflowastffromtensorflow.keras.preprocessing.imageimportImageDataGenerator# 设置路径train_dircat_dog_dataset/trainvalidation_dircat_dog_dataset/validation# 图像生成器train_datagenImageDataGenerator(rescale1./255,rotation_range40,width_shift_range0.2,height_shift_range0.2,shear_range0.2,zoom_range0.2,horizontal_flipTrue,fill_modenearest)validation_datagenImageDataGenerator(rescale1./255)train_generatortrain_datagen.flow_from_directory(train_dir,target_size(150,150),batch_size20,class_modebinary)validation_generatorvalidation_datagen.flow_from_directory(validation_dir,target_size(150,150),batch_size20,class_modebinary)# 构建CNN模型modeltf.keras.models.Sequential([tf.keras.layers.Conv2D(32,(3,3),activationrelu,input_shape(150,150,3)),tf.keras.layers.MaxPooling2D(2,2),tf.keras.layers.Conv2D(64,(3,3),activationrelu),tf.keras.layers.MaxPooling2D(2,2),tf.keras.layers.Conv2D(128,(3,3),activationrelu),tf.keras.layers.MaxPooling2D(2,2),tf.keras.layers.Conv2D(128,(3,3),activationrelu),tf.keras.layers.MaxPooling2D(2,2),tf.keras.layers.Flatten(),tf.keras.layers.Dense(512,activationrelu),tf.keras.layers.Dense(1,activationsigmoid)])model.compile(lossbinary_crossentropy,optimizertf.keras.optimizers.RMSprop(lr1e-4),metrics[accuracy])historymodel.fit(train_generator,steps_per_epoch100,epochs30,validation_datavalidation_generator,validation_steps50)model.save(cats_and_dogs_small_1.h5)4. 创建PyQt5用户界面接下来我们将编写代码来集成训练好的模型与PyQt5以实现用户界面。编写main.pyimportsysfromPyQt5.QtWidgetsimportQApplication,QMainWindow,QPushButton,QFileDialog,QLabel,QVBoxLayout,QWidgetfromPyQt5.QtGuiimportQPixmapimportcv2importnumpyasnpfromtensorflow.keras.modelsimportload_modelclassCatDogDetectionApp(QMainWindow):def__init__(self):super().__init__()self.initUI()self.modelload_model(cats_and_dogs_small_1.h5)definitUI(self):self.setWindowTitle(猫狗动物识别系统)self.setGeometry(100,100,800,600)self.image_labelQLabel(self)self.image_label.setAlignment(Qt.AlignCenter)btn_imageQPushButton(选择图片,self)btn_image.clicked.connect(self.openImage)layoutQVBoxLayout()layout.addWidget(self.image_label)layout.addWidget(btn_image)containerQWidget()container.setLayout(layout)self.setCentralWidget(container)defopenImage(self):optionsQFileDialog.Options()file_name,_QFileDialog.getOpenFileName(self,选择图片,,Images (*.png *.xpm *.jpg);;All Files (*),optionsoptions)iffile_name:self.detectAndShow(file_name)defdetectAndShow(self,image_path):imgcv2.imread(image_path)imgcv2.resize(img,(150,150))img_arraynp.expand_dims(img,axis0)predictionself.model.predict(img_array)labelCatifprediction0.5elseDogresult_imgcv2.putText(img,label,(10,30),cv2.FONT_HERSHEY_SIMPLEX,1,(255,0,0),2)height,width,channelresult_img.shape bytes_per_line3*width q_imgQPixmap.fromImage(QImage(result_img.data,width,height,bytes_per_line,QImage.Format_RGB888).rgbSwapped())self.image_label.setPixmap(q_img)if__name____main__:appQApplication(sys.argv)exCatDogDetectionApp()ex.show()sys.exit(app.exec_())5. 运行项目确保所有依赖项已正确安装后你可以运行上述脚本python main.py解释初始化CNN模型通过load_model加载预训练的猫狗分类模型。创建PyQt5界面使用QMainWindow和QWidget构建主窗口和布局。添加按钮用于选择图片并显示检测结果。处理图片当用户选择一张图片时调整图片大小到模型输入尺寸150x150并进行预测。根据预测结果在图片上绘制类别标签并显示在界面上。增加视频支持为了增加对视频的支持可以在界面中添加相应的按钮并编写对应的处理逻辑。添加视频功能defopenVideo(self):optionsQFileDialog.Options()file_name,_QFileDialog.getOpenFileName(self,选择视频,,Videos (*.mp4 *.avi);;All Files (*),optionsoptions)iffile_name:self.detectVideo(file_name)defdetectVideo(self,video_path):capcv2.VideoCapture(video_path)whilecap.isOpened():ret,framecap.read()ifnotret:breakframecv2.resize(frame,(150,150))img_arraynp.expand_dims(frame,axis0)predictionself.model.predict(img_array)labelCatifprediction0.5elseDogresult_framecv2.putText(frame,label,(10,30),cv2.FONT_HERSHEY_SIMPLEX,1,(255,0,0),2)cv2.imshow(Video Detection,result_frame)ifcv2.waitKey(1)0xFFord(q):breakcap.release()cv2.destroyAllWindows()
