我是机器学习的新手,可能需要一些有关如何进行项目/问题的常规提示/指南。
我想将转移学习与mobilenetV2结合使用,对5种不同类型的火车(5类)进行分类。从简单开始,我只有2个班级,并且在冻结所有基本层的情况下,可以达到〜85%的训练精度和〜70%的验证精度。大约10个时代之后,这还不错。.但是:
主要问题是:一旦我增加了火车班的数量,准确性就不会提高到38%以上。我尝试更改学习率,批量大小,图像输入形状..但没有成功...
由于我是该领域的新手,使用常规设置也可能无法解决问题...
我的想法:
您可以在Jupyter笔记本中找到以下代码:https://github.com/n304win/transferLearn
非常感谢帮助! 谢谢n304win