search

naturomics / CapsLayer

CapsLayer: An advanced library for capsule theory
Apache License 2.0
361 stars 116 forks source link

ModuleNotFoundError: No module named 'capslayer.data.datasets.stanford_drone' #38

Open akorez opened 5 years ago

akorez commented 5 years ago

Hi, I created the datareader and datawriter according to my own dataset by following the tutorials. I made the necessary changes to config.py and main.py files. But I get an error as mentioned in the title. How can I create my own module similar to the cifar10 module? Please, help me! Thanks...

naturomics commented 5 years ago

python can not find your dataloader, you should import your module to this init.py file

akorez commented 5 years ago

I added my own module into the init.py file. Here's the code. I still get the same error.

from future import absolute_import from future import division from future import print_function

from . import mnist from . import fashion_mnist from . import cifar10 from . import cifar100 from . import stanford_drone

all = ['mnist', 'fashion_mnist', 'cifar10', 'cifar100','stanford_drone']