我是否必须将每个测试图像与每个类的示例图像进行比较?测试集包含104个类别的大约7400张图像。所以这将是7400 x 104的预测?
在tpu上使用tensorflow,我能够非常有效地训练模型。但是,使用上述方法预测标签会花费很长时间,此外,模型预测调用还会导致内存泄漏,最终导致内核失败(内存可能高达30 + gb并计数)。
您可以通过多种方式执行此操作:
如果您对KNN不太了解或不想看代码实现,也可以参考Blog。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句