CamVid dataset??

[shubhamk-76]

I trie to implement this repository using CamVid dataset but was getting some error. The error are relate to class as there are 32 classes in CamVid dataset.

File "/mnt/DATA/CS24S001/pytorch-segmentation/base/base_trainer.py", line 101, in train
results = self._train_epoch(epoch)
File "/mnt/DATA/CS24S001/pytorch-segmentation/trainer.py", line 106, in _train_epoch
seg_metrics = self._get_seg_metrics()
File "/mnt/DATA/CS24S001/pytorch-segmentation/trainer.py", line 204, in _get_seg_metrics
"Class_IoU": dict(zip(range(self.num_classes), np.round(IoU, 3)))
TypeError: 'numpy.float64' object is not iterable