Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【idea】keras 模型优化 #147

Open
williamfzc opened this issue Nov 2, 2020 · 0 comments
Open

【idea】keras 模型优化 #147

williamfzc opened this issue Nov 2, 2020 · 0 comments
Labels
help wanted Extra attention is needed

Comments

@williamfzc
Copy link
Owner

williamfzc commented Nov 2, 2020

目前这个工程使用的keras模型基本参照着keras官方文档中对于小型训练集的设计来的(猫狗大战,百级别)
https://keras-cn.readthedocs.io/en/latest/legacy/blog/image_classification_using_very_little_data/

  • 但实际上,通常来说我们使用的数据集要比传统的小数据集还要小很多,而且需要支持多分类;
  • 这也导致了在部分情况下模型的适应性较差,而一味增加epochs并不一定能解决问题;
  • 在这种情况下,我们希望我们内置的模型是足以满足大部分常见场景的、具备更强通用性的;

从现在收集到的数据来看:

  • 3-4类别
  • 训练集每类别 10-20 张

https://github.com/williamfzc/stagesepx/blob/master/stagesepx/classifier/keras.py
你可以通过继承 KerasClassifier 后修改里面的 模型设计部分来改善模型

一些相关链接(持续补充到这里):

thanks!

@williamfzc williamfzc added the help wanted Extra attention is needed label Nov 2, 2020
@williamfzc williamfzc pinned this issue Nov 2, 2020
@williamfzc williamfzc changed the title keras 模型优化 【idea】keras 模型优化 May 24, 2021
@williamfzc williamfzc unpinned this issue May 4, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

1 participant