28 Star 128 Fork 25

cungudafa / hand-keras-yolo3-recognize

Clone or download
yolo_video.py
Copy Edit Web IDE Raw Normal History
cungudafa   sign 2020-08-08
import sysimport argparsefrom yolo import YOLO, detect_videofrom PIL import Imagedef detect_img(yolo):    while True:        print("Press 'q' quit !")        img = input('Input image filename:')        if img.lower() == 'q':            exit(0)        try:            image = Image.open(img)        except:            print('Open Error! Try again!')            continue        else:                        r_image = yolo.detect_image(image)            r_image.show()    yolo.close_session()FLAGS = Noneif __name__ == '__main__':    # class YOLO defines the default value, so suppress any default here    parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)    '''    Command line options    '''    parser.add_argument(        '--model_path', type=str,        help='path to model weight file, default ' + YOLO.get_defaults("model_path")    )    parser.add_argument(        '--anchors_path', type=str,        help='path to anchor definitions, default ' + YOLO.get_defaults("anchors_path")    )    parser.add_argument(        '--classes_path', type=str,        help='path to class definitions, default ' + YOLO.get_defaults("classes_path")    )    parser.add_argument(        '--gpu_num', type=int,        help='Number of GPU to use, default ' + str(YOLO.get_defaults("gpu_num"))    )    parser.add_argument(        '--image', default=False, action="store_true",        help='Image detection mode, will ignore all positional arguments'    )    '''    Command line positional arguments -- for video detection mode    '''    parser.add_argument(        "--input", nargs='?', type=str,required=False,default='./path2your_video',        help = "Video input path"    )    parser.add_argument(        "--output", nargs='?', type=str, default="",        help = "[Optional] Video output path"    )    FLAGS = parser.parse_args()    if FLAGS.image:        """        Image detection mode, disregard any remaining command line arguments        """        print("Image detection mode")        if "input" in FLAGS:            print(" Ignoring remaining command line arguments: " + FLAGS.input + "," + FLAGS.output)        detect_img(YOLO(**vars(FLAGS)))    elif "input" in FLAGS:        detect_video(YOLO(**vars(FLAGS)), FLAGS.input, FLAGS.output)    else:        print("Must specify at least video_input_path.  See usage with --help.")
Python
1
https://gitee.com/cungudafa/hand-keras-yolo3-recognize.git
git@gitee.com:cungudafa/hand-keras-yolo3-recognize.git
cungudafa
hand-keras-yolo3-recognize
hand-keras-yolo3-recognize
master

Search