calculate_class_weighting.py

[ThomasLengeling]

Hi I am following your tutorial to train on the Cityscape dataset.

I am running the following code for the ENet predictions.

python calculate_class_weighting.py --source ENet/dataset/train_fine_cityscapes.txt --num_classes 19 

but i get:

Traceback (most recent call last):
  File "calculate_class_weighting.py", line 48, in <module>
    if ((labels == i).sum()) == 0:
AttributeError: 'bool' object has no attribute 'sum'

thank you

[pedropgusmao]

@ThomasLengeling , I changed line 48 it to if ( np.sum( (labels == i) ) ) == 0: and it works

[TimoSaemann]

Thanks @pedropgusmao