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

Question

我是深度学习及其框架的新手。我收集了一些自行车和自行车物体的jpg图像。现在，我想使用胶囊网络训练这些图像。

胶囊网络是在mnist数据集上实现的。代码为here。

应用我的自定义数据集需要进行哪些更改。图片应该更改为numpy格式还是tfrecord格式？以及如何将它们加载到网络上？

Answer 1

要加载自定义数据集，我引用了this。它有助于。数据集将转换为numpy数组，并使用了keras背景。

我加载了自定义数据集，对于胶囊网络模型，请使用this。