YOLO V2 predicting one class

Situation - Detecting 1 class object

• User is performing transfer learning YOLO V2 model with own data.
• User is trying to detect one class of object.

Solution

• User need to provide 2 classes for Object with or Without classes.
• If data has more than one classes, no need to define extra class.

How to troubleshoot

1. Inspect the root cause of problem. In this case String labels = label.get(obj.getPredictedClass()); result in array out of bound.
2. Inspect number of classes. Having number of class = 1 is the cause of problem.