如何解决CreateML mlModel 无法推断几个类
我有一个图像分类问题(黑白简笔画) 问题是永远不会推断出两个类。
训练数据集有 500 个类,每个类有 100 个或更多 (299x299,1) 个样本。无法识别的类已从 100 个样本增加到 140+ 个样本。最近,我使用 autosplit 训练了 CreateML,但使用了单独生成的验证数据集。结果没有区别。 CoreML 内部的“FeaturePrint”模型似乎只是拒绝识别这两个字形。
值得注意的是,一个简单的 TensorFlow/Keras 模型(8 层 CNN/ANN)适用于所有类别。但是,tf 模型在未经训练的“野生”输入图像上的性能较差,转换为 ML 格式进行部署时的 tf 模型为 160 MB,而 CreateML 模型为 8 MB。
对处理 CreateML 有什么建议吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。