search

LikeLy-Journey / SegmenTron

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)
Apache License 2.0
705 stars 162 forks source link

erorr in cityscapes #10

Closed bj5546 closed 4 years ago

bj5546 commented 4 years ago

in cityscapes datasets，a error happened . assert: DataLoader worker process 0 in line 57 assert (value in self._mapping). I changed it in line 51 to self._mapping = np.array(range(0, len(self._key))).astype('int32'). It works! Is it right?

LikeLy-Journey commented 4 years ago

in cityscapes datasets，a error happened . assert: DataLoader worker process 0 in line 57 assert (value in self._mapping). I changed it in line 51 to self._mapping = np.array(range(0, len(self._key))).astype('int32'). It works! Is it right?

you use your custom data? if you use cityscape, you do not need to change anything.