MendelXu / SAN

Open-vocabulary Semantic Segmentation
https://mendelxu.github.io/SAN/
MIT License
295 stars 27 forks source link

数据集问题 #54

Closed runhuzhao closed 4 months ago

runhuzhao commented 4 months ago

-- Process 1 terminated with the following error: Traceback (most recent call last): File "/home/data/anaconda3/envs/san/lib/python3.9/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap fn(i, args) File "/home/data/anaconda3/envs/san/lib/python3.9/site-packages/detectron2/engine/launch.py", line 126, in _distributed_worker main_func(args) File "/home/data/project/SAN/train_net.py", line 271, in main trainer = Trainer(cfg) File "/home/data/anaconda3/envs/san/lib/python3.9/site-packages/detectron2/engine/defaults.py", line 378, in init data_loader = self.build_train_loader(cfg) File "/home/data/project/SAN/train_net.py", line 94, in build_train_loader return build_detection_train_loader(cfg, mapper=mapper) File "/home/data/anaconda3/envs/san/lib/python3.9/site-packages/detectron2/config/config.py", line 207, in wrapped explicit_args = _get_args_from_config(from_config, *args, *kwargs) File "/home/data/anaconda3/envs/san/lib/python3.9/site-packages/detectron2/config/config.py", line 245, in _get_args_from_config ret = from_config_func(args, **kwargs) File "/home/data/project/SAN/san/data/build.py", line 161, in _train_loader_from_config dataset = get_detection_dataset_dicts( File "/home/data/project/SAN/san/data/build.py", line 129, in get_detection_dataset_dicts assert len(dicts), "Dataset '{}' is empty!".format(dataset_name) AssertionError: Dataset 'coco_2017_train_stuff_sem_seg' is empty!

求助一下这个错误是因为什么呀!

runhuzhao commented 4 months ago

麻烦您啦,我已经找到问题了,因为在register_coco_stuff_164k.py文件中,读取图片是按jpg格式读取的;而我用Simseg中的方法划分数据集的时候,生成的图片是png格式,所以没有读取到数据,报错提示Dataset 为空。