eric-yyjau/pytorch-superpoint

Loader.py error

eminerkol-60 opened this issue · 1 comments

Hi, I am currently trying to use synthetic dataset for the homography adaptation task and I am getting the error below;
Traceback (most recent call last):
File "export.py", line 409, in
args.func(config, output_dir, args)
File "/home/hmahmad/.local/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "export.py", line 243, in export_detector_homoAdapt_gpu
data = dataLoader(config, dataset=task, export_task=export_task)
File "/home/hmahmad/Desktop/Emin/pytorch-superpoint-master/utils/loader.py", line 105, in dataLoader_test
from datasets.SyntheticDataset import SyntheticDataset
ImportError: cannot import name 'SyntheticDataset'
there is no file called SyntheticDataset inside of the datasets folder. any help is appreciated.

Thanks.

Hi @eminerkol-60,

I think there's a bug there.
If I remember correctly, you'll train magicPoint from synthetic dataset with this config.

dataset: 'SyntheticDataset_gaussian'

This will import dataload from SyntheticDataset_gaussian instead of SyntheticDataset.
The error and code you show should be cleaned up.
If you have an update, feel free to submit a pull request.

Thanks.