如何将自定义数据集加载到胶囊网络？

Question

我是深入学习及其框架的新手。我收集了一些jpg图片的自行车和自行车的对象。现在，我想用胶囊网络来训练这些图像。你知道吗

胶囊网络是在mnist数据集上实现的。代码是here。你知道吗

应用自定义数据集需要做哪些更改。图像应该更改为numpy格式还是tfrecord格式？如何在网络上加载它们？你知道吗

Answer 1

要加载自定义数据集，我引用了this。这很有帮助。将数据集转换为numpy数组，并使用keras background。你知道吗

我加载自定义数据集，对于胶囊网络模型，使用this。你知道吗