我正在尝试为自己的包含jpg图片的自定义数据集实现https://github.com/naturomics/CapsLayer。我已经将这些图像转换为tfrecord格式。在main.py文件中需要进行哪些强制更改？
我已经更改了定义的更改，例如修改参数（高度，宽度，通道等）。但是我仍然对如何导入DataLoader对象感到困惑。 以下是有关应用自定义数据集的说明：https://github.com/naturomics/CapsLayer/blob/master/docs/tutorials.md 给定的代码针对mnist数据集运行，但如果使用任何其他数据集，则它将显示main.py文件中第221行的ModuleNotFound错误。