search

IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.
MIT License
86 stars 19 forks source link

I met this error when run. #4

Closed bemoregt closed 4 years ago

bemoregt commented 4 years ago

Hi, @IanTaehoonYoo

I met this error when run.

AssertionError Traceback (most recent call last)

in 31 ]) 32 ---> 33 train_datasets = SegmentationDataset(train_images, train_labled, n_classes, compose) 34 train_loader = torch.utils.data.DataLoader(train_datasets, batch_size=batch_size, shuffle=True, drop_last=True) 35 ~/semantic-segmentation-pytorch/segmentation/data_loader/segmentation_dataset.py in __init__(self, images_dir, segs_dir, n_classes, transform) 51 self.pairs_dir = self._get_image_pairs_(self.images_dir, self.segs_dir) 52 verified = self._verify_segmentation_dataset() ---> 53 assert verified 54 55 def __len__(self): AssertionError: ----------------------------------------------- My Data have RGB images and Grayscale 1class labeled. and have same filename and same size. What's wrong to me ? Thanks. @bemoregt.