ykdai / Flare7K

Official Implementation of "Flare7K: A Phenomenological Nighttime Flare Removal Dataset"
https://ykdai.github.io/projects/Flare7K
Other
119 stars 16 forks source link

Can I get your help on code execution? #6

Closed staraaaa closed 1 year ago

staraaaa commented 1 year ago

Excuse me, I always get the following error when debugging code locally according to your requirements

Name Flare_Pair_Loader is not found, use name: Flare_Pair_Loader_basicsr! Traceback (most recent call last): File "basicsr/train.py", line 215, in <module> train_pipeline(root_path) File "basicsr/train.py", line 120, in train_pipeline result = create_train_val_dataloader(opt, logger) File "basicsr/train.py", line 35, in create_train_val_dataloader train_set = build_dataset(dataset_opt) File "/home/cvgroup/anaconda3/envs/flare/lib/python3.7/site-packages/basicsr/data/init.py", line 34, in build_dataset dataset = DATASET_REGISTRY.get(dataset_opt['type'])(dataset_opt) File "/home/cvgroup/anaconda3/envs/flare/lib/python3.7/site-packages/basicsr/utils/registry.py", line 71, in get raise KeyError(f"No object named '{name}' found in '{self._name}' registry!") KeyError: "No object named 'Flare_Pair_Loader' found in 'dataset' registry!" If I could get your help, I would be extremely grateful!

ykdai commented 1 year ago

Is there a flare7k_dataset.py in your 'home/cvgroup/anaconda3/envs/flare/lib/python3.7/site-packages/basicsr/data/' path? Seems you load the basicsr from the anaconda path, and not from the project path.

ykdai commented 1 year ago

Hello, I have updated setup files in the root directory. If you still have not fixed this problem, you may try the Installation method in README.

staraaaa commented 1 year ago

Thank you very much for your help. It means a lot to me!