不平衡和小的图像分类数据集

Question

我有两个图像分类数据集：

• 损坏的巧克力包装有27张图像
• 未损坏的161张巧克力包装

我必须编写一个python分类器来区分警报产生。

解决我的问题的最佳方法是什么？带有转移学习或K均值解决方案的CNN？

Answer 1

似乎您的数据很少。对我而言，转学是解决此问题的最佳方法。但是，请尝试查找更多数据或使用数据扩充功能（您可以阅读this）。

此外，如果您正在寻找CNN的转学，我想this tutorial可以为您提供帮助。