sisyphus$ python flow --model cfg/yolo-voc-6c.cfg --load -1 --demo camera --json
import sys
from darkflow.cli import cliHandler
import darkflow.net.yolov2.predict
def Predict_Image(imgdir_):
sys.argv = ['flow','--imgdir',imgdir_,'--model','cfg/yolo-voc-6c.cfg','--load','-1','--json']
cliHandler(sys.argv)
result=darkflow.net.yolov2.predict.postprocess()
print(result)
def Predict_Camera():
sys.argv = ['flow','--model','cfg/yolo-voc-6c.cfg','--load','-1','--demo','camera','--json']
cliHandler(sys.argv)
if __name__=='__main__':
imgdir='sample_img_test1'
Predict_Image(imgdir)
没有评论:
发表评论