小伍同学的博客

本文介绍了如何使用PyTorch和PyQt5实现一个简单的猫狗分类器。该分类器基于Kaggle的DogsvsCats数据集，包含25000张训练图片和12500张测试图片。文章详细描述了如何构建一个三层卷积神经网络（CNN）模型，并通过PyQt5设计了一个交互界面，用户可以通过该界面加载图片并查看分类结果。代码分为两部分：cat_dog_train.py用于训练模型，cat_dog_test.py用于测试和展示分类结果。训练过程中使用了数据增强和标准化技术，并保存了模型检查点。测试部分通过加载训练好的模型，