从github上下载了keras-yolo3项目代码,体验了一下,感觉还不错,代码全是用python写的,字面语句很好理解。
作为一个简单的体验,我写了一个小小的测试程序,打开笔记本摄像头,实时使用YOLO检测。
编写文件 usb-camera.py,并把文件放在keras-yolo3 目录下面。
# -*- coding: utf-8 -*-
"""
Created on Tue May 7 15:07:52 2019
@author: vector
"""
import cv2
import sys
import argparse
from yolo import YOLO, detect_video
from PIL import Image
import numpy as np
#这一段抄的yolo-video.py中的函数
def CreateYOLO():
# 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', type=str,
help='path to model weight file, default ' + YOLO.get_defaults("model_path")
)
parser.add_argument(
'--anchors', type=str,