fpv-iplab / MECCANO

The MECCANO Dataset: official repository in which we provide code and models.
32 stars 2 forks source link

No object named 'Meccano' found in 'DATASET' registry #11

Open aranciokov opened 1 year ago

aranciokov commented 1 year ago

Hi, thanks for releasing this interesting project. I'm trying to work on the action recognition task. So far, I have done the following:

Now, to the problem. When I execute "python tools/run_net.py --cfg ../configs/action_recognition/SLOWFAST_8x8_R50_MECCANO.yaml" I get the following error:

Traceback (most recent call last): File "/media/HDD/alexfalcon/MECCANO/env_meccano/lib/python3.8/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap fn(i, *args) File "/media/HDD/alexfalcon/MECCANO/slowfast/slowfast/utils/multiprocessing.py", line 60, in run ret = func(cfg) File "/media/HDD/alexfalcon/MECCANO/slowfast/tools/train_net.py", line 596, in train train_loader = loader.construct_loader(cfg, "train") File "/media/HDD/alexfalcon/MECCANO/slowfast/slowfast/datasets/loader.py", line 114, in construct_loader dataset = build_dataset(dataset_name, cfg, split) File "/media/HDD/alexfalcon/MECCANO/slowfast/slowfast/datasets/build.py", line 31, in build_dataset return DATASET_REGISTRY.get(name)(cfg, split) File "/media/HDD/alexfalcon/MECCANO/env_meccano/lib/python3.8/site-packages/fvcore/common/registry.py", line 71, in get raise KeyError( KeyError: "No object named 'Meccano' found in 'DATASET' registry!"

Unfortunately, I can't seem to figure out the cause of this issue. Any idea? Thanks and cheers, Alex

francescoragusa commented 1 year ago

Hi @aranciokov, thank you for your interest about our work!

It seems strange that the error that you described occurs. To avoid that error, it is fundamental to replace the __init__.py file in the "slowfast/datasets/" folder where the Meccano class is imported:

from .meccano import Meccano #added

Can you check that in your /slowfast/datasets/__init__.py file there is that "import"?