Solved by chhaya_kumar_das in post #8 
              
              
                I have one question. In the  num_classes  argument do we have to include Background also as a class. Because originally I have 4 classes. When I run with 5 classes (including background)  it does not throw me any error.