How can I make changes to make it work for only 5 classes.

[gangooteli]

I need to do detection for only 5 classes. How to work on that. I tried to make changes in yolov2-coco.meta file and reduce it to 5 classes. But It is giving me error.

Forwarding the image input. Cost time of run = 1.40356898308s. Traceback (most recent call last): File "demo_yolo_v2.py", line 203, in demo_image() File "demo_yolo_v2.py", line 157, in demo_image results = yolo.detect_object(im) File "demo_yolo_v2.py", line 143, in detect_object result = self.postprocess(out[0]) File "demo_yolo_v2.py", line 124, in postprocess result = box.box_constructor(meta,net_out) File "/home/nikhil/Documents/yolo/yolov2-tensorflow-master/utils/box.py", line 68, in box_constructor class_probs = np.ascontiguousarray(Classes).reshape([HWB, C]) ValueError: cannot reshape array of size 144400 into shape (1805,5)