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)