import sys, os
import cv2
import datetime
sys.path.append(os.path.join(os.getcwd(),'python/'))

import darknet as dn
import pdb

dn.set_gpu(0)
net = dn.load_net("VOCdevkit/VOC2009/yolov3-voc-sgdt-hat.cfg", "backup/yolov3-voc-sgdt-hat_last.weights", 0)
meta = dn.load_meta("VOCdevkit/VOC2009/voc-hat.data")

vc = cv2.VideoCapture('video/V3.mp4')

rval = vc.isOpened()

c=0

while rval:

    start = datetime.datetime.now()
    rval, frame = vc.read()

    if rval:
        if(c>9999):
            break

        cv2.imwrite("video/"+str(c) + '.jpg', frame) #i

        r = dn.detect(net, meta, "video/"+str(c) + '.jpg')
        print len(r)
        os.remove("video/"+str(c) + '.jpg')
        c=c+1
        
    else:
        break
    end = datetime.datetime.now()
    print end - start

vc.release()