IDEA-Research / MaskDINO

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AssertionError

sym330 opened this issue · comments

When I run
export DETECTRON2_DATASETS=/media/user/b68b12ae-2e14-474d-971b-810bc8fd28d5/MSY/ade20k_crop/
python train_net.py --num-gpus 1 --config-file configs/ade20k/semantic-segmentation/Base-ADE20K-SemanticSegmentation.yaml MODEL.WEIGHTS maskdino_r50_50ep_100q_celoss_hid1024_3s_semantic_ade20k_48.7miou.pth

The code will report errors AssertionError

File "/media/user/b68b12ae-2e14-474d-971b-810bc8fd28d5/MSY/MaskDINO-main/maskdino/data/dataset_mappers/mask_former_semantic_dataset_mapper.py", line 126, in call
aug_input, transforms = T.apply_transform_gens(self.tfm_gens, aug_input)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/augmentation.py", line 363, in apply_augmentations
tfms = inputs.apply_augmentations(augmentations)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/augmentation.py", line 350, in apply_augmentations
return AugmentationList(augmentations)(self)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/augmentation.py", line 267, in call
tfm = x(aug_input)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/augmentation.py", line 173, in call
aug_input.transform(tfm)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/augmentation.py", line 342, in transform
self.sem_seg = tfm.apply_segmentation(self.sem_seg)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/transform.py", line 155, in apply_segmentation
segmentation = self.apply_image(segmentation, interp=Image.NEAREST)
File "/home/user/anaconda3/envs/ldm/lib/python3.8/site-packages/detectron2/data/transforms/transform.py", line 113, in apply_image
assert img.shape[:2] == (self.h, self.w)
AssertionError