The issue with the dataset during the training process.

[jixiaofeng123]

Hello, when I run the Python script datasets/prepare_coco_stuff_sem_seg.py to generate a dataset, it shows 0it [00:00, ?it/s], and when I execute python train_net.py --num-gpu 1 --config-file configs/ovseg_swinB_vitL_bs32_120k.yaml MODEL.CLIP_ADAPTER.MASK_PROMPT_FWD False, I encounter the following error.
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
relative_position_bias_table
Traceback (most recent call last):
File "train_net.py", line 302, in
launch(
File "/usr/local/lib/python3.8/dist-packages/detectron2/engine/launch.py", line 82, in launch
main_func(*args)
File "train_net.py", line 294, in main
trainer = Trainer(cfg)
File "/usr/local/lib/python3.8/dist-packages/detectron2/engine/defaults.py", line 378, in init
data_loader = self.build_train_loader(cfg)
File "train_net.py", line 106, in build_train_loader
return build_detection_train_loader(cfg, mapper=mapper, dataset=dataset)
File "/usr/local/lib/python3.8/dist-packages/detectron2/config/config.py", line 207, in wrapped
explicit_args = _get_args_from_config(from_config, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/detectron2/config/config.py", line 245, in _get_args_from_config
ret = from_config_func(*args, **kwargs)
File "/root/ov-seg/open_vocab_seg/data/build.py", line 164, in _train_loader_from_config
dataset = get_detection_dataset_dicts(
File "/root/ov-seg/open_vocab_seg/data/build.py", line 127, in get_detection_dataset_dicts
dataset_dicts = [
File "/root/ov-seg/open_vocab_seg/data/build.py", line 128, in
wrap_metas(DatasetCatalog.get(dataset_name), dataset_name=dataset_name)
File "/usr/local/lib/python3.8/dist-packages/detectron2/data/catalog.py", line 58, in get
return f()
File "/root/ov-seg/open_vocab_seg/data/datasets/register_coco_stuff.py", line 235, in
lambda x=image_dir, y=gt_dir: load_sem_seg(
File "/usr/local/lib/python3.8/dist-packages/detectron2/data/datasets/coco.py", line 274, in load_sem_seg
assert len(gt_files) > 0, "No annotations found in {}.".format(gt_root)
AssertionError: No annotations found in datasets/coco/stuffthingmaps_detectron2/train2017.

[Jeff-LiangF]

Hi @jixiaofeng123 ,

Looks like there was some error with the data preparation. Could you double-check the data structure?