DEV Community
·
机器学习中的二分类：神经网络用于分类吉娃娃与松饼
💡
原文英文，约1000词，阅读约需4分钟。
📝
内容提要
图像分类是机器学习的基本任务，旨在将图像进行分类。传统方法依赖局部特征，而深度学习通过卷积神经网络（CNN）自动提取特征。本文探讨了在“吉娃娃与松饼”数据集上进行二分类的挑战，比较了多层神经网络（MLNN）和CNN的性能。经过数据清理和预处理，CNN的表现优于MLNN，数据增强进一步提升了模型效果。未来将研究更复杂的架构和更大数据集。
🎯
关键要点
- 图像分类是机器学习的基本任务，旨在将图像进行分类。
- 传统方法依赖局部特征，而深度学习通过卷积神经网络（CNN）自动提取特征。
- 本文探讨了在“吉娃娃与松饼”数据集上进行二分类的挑战。
- 比较了多层神经网络（MLNN）和CNN的性能，发现CNN表现优于MLNN。
- 经过数据清理和预处理，数据增强进一步提升了模型效果。
- 数据集包含3199张吉娃娃图像和2718张松饼图像。
- 数据清理过程中移除了错误标记和无关图像，确保数据集的相关性。
- 数据集的失衡比率为1.22，表明没有显著失衡。
- 预处理步骤包括图像调整大小、裁剪、归一化和数据增强。
- 创建了三种数据集变体：原始RGB图像、增强RGB图像和分割RGB图像。
- MLNN模型由全连接层、批归一化、丢弃层和激活函数组成。
- CNN模型更适合处理空间数据，包含卷积层、池化层和分类层。
- TogNet是一个定制的CNN，旨在解决过拟合问题。
- 研究结果表明，CNN，特别是TogNet架构，在二分类图像分类中优于MLNN。
- 未来的工作将探索更复杂的架构和更大数据集，以进一步提高分类准确性。
➡️
