search

allenai / satlas-super-resolution

Apache License 2.0
190 stars 24 forks source link

SSRDataset is not found! #14

Closed yunseok624 closed 5 months ago

yunseok624 commented 5 months ago

Hi, I'm planning to run esrgan_s2naip_urban.yml, but I encountered one problem:

Name SSRDataset is not found, use name: SSRDataset_basicsr! Traceback (most recent call last): File "/trinity/shared/opt/python/3.8.5/lib/python3.8/runpy.py", line 194, in _run_module_as_main return _run_code(code, main_globals, None, File "/trinity/shared/opt/python/3.8.5/lib/python3.8/runpy.py", line 87, in _run_code exec(code, run_globals) File "/beegfs/home/park.yunseok/satlas-super-resolution/ssr/train.py", line 182, in train_pipeline(root_path) File "/beegfs/home/park.yunseok/satlas-super-resolution/ssr/train.py", line 80, in train_pipeline result = create_train_val_dataloader(opt, logger) File "/trinity/home/park.yunseok/.local/lib/python3.8/site-packages/basicsr/train.py", line 35, in create_train_val_dataloader train_set = build_dataset(dataset_opt) File "/trinity/home/park.yunseok/.local/lib/python3.8/site-packages/basicsr/data/init.py", line 34, in build_dataset dataset = DATASET_REGISTRY.get(dataset_opt['type'])(dataset_opt) File "/trinity/home/park.yunseok/.local/lib/python3.8/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 'SSRDataset' found in 'dataset' registry!"

How can I solve such problem? Thank you in advance

piperwolters commented 5 months ago

Hi, it looks like that config is using the old naming scheme for the dataset. I pushed a change to that config that replaces "SSRDataset" with "S2NAIPDataset". Other config files look fine.