用脚本代替flow

import sys
from darkflow.cli import cliHandler
import darkflow.net.yolov2.predict

'''
python flow --model cfg/yolo-voc-6c.cfg --load bin/yolo.weights
 --train --annotation /Users/sisyphus/darkflow/VOCtest2018/Annotation
 --dataset /Users/sisyphus/darkflow/VOCtest2018/JPEGImages --gpu 1.0
 '''

def Train():
    sys.argv = ['flow', '--model', 'cfg/yolo-voc-6c.cfg', '--load', 'bin/yolo.weights', '--train']
    cliHandler(sys.argv)
   

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__':
    Train()


#    imgdir='sample_img_test30'
#    Predict_Image(imgdir)
   
#    Predict_Camera()