search

PingoLH / FCHarDNet

Fully Convolutional HarDNet for Segmentation in Pytorch
MIT License
195 stars 52 forks source link

Train sunrgbd data set to report errors #23

Open 18022443868 opened 4 years ago

18022443868 commented 4 years ago

149it [01:19, 1.85it/s][0 1 2 5 7] [0 1 2 5 7] [ 0 1 2 5 14] [ 0 1 2 5 14] [ 0 1 2 5 8 9 14] [ 0 1 2 5 9 14] Traceback (most recent call last): File "E:/vSLAM/FCHarDNet-master/train.py", line 269, in train(cfg, writer, logger) File "E:/vSLAM/FCHarDNet-master/train.py", line 183, in train for i_val, (images_val, labels_val) in tqdm(enumerate(valloader)): File "D:\Anaconda3\envs\pytorch1\lib\site-packages\tqdm\std.py", line 1102, in iter for obj in iterable: File "D:\Anaconda3\envs\pytorch1\lib\site-packages\torch\utils\data\dataloader.py", line 560, in next batch = self.collate_fn([self.dataset[i] for i in indices]) File "D:\Anaconda3\envs\pytorch1\lib\site-packages\torch\utils\data\dataloader.py", line 560, in batch = self.collate_fn([self.dataset[i] for i in indices]) File "E:\vSLAM\FCHarDNet-master\ptsemseg\loader\sunrgbd_loader.py", line 81, in getitem img, lbl = self.transform(img, lbl) File "E:\vSLAM\FCHarDNet-master\ptsemseg\loader\sunrgbd_loader.py", line 106, in transform assert np.all(classes == classes1) AssertionError

Training sunrgbd data set error why is the category not correct here?Is floating-point 64 changed to 32 by default?