42 lines
1011 B
Python
42 lines
1011 B
Python
|
import time
|
|||
|
|
|
|||
|
|
import cv2
|
|||
|
|
import numpy as np
|
|||
|
|
from PIL import Image
|
|||
|
|
|
|||
|
|
from yolo import YOLO
|
|||
|
|
|
|||
|
|
yolo = YOLO()
|
|||
|
|
|
|||
|
|
capture = cv2.VideoCapture(0)
|
|||
|
|
# 1 就是外接摄像头 0 就是自己的摄像头
|
|||
|
|
ref, frame = capture.read()
|
|||
|
|
fps = 0.0
|
|||
|
|
while (True):
|
|||
|
|
t1 = time.time()
|
|||
|
|
# 读取某一帧
|
|||
|
|
ref, frame = capture.read()
|
|||
|
|
if not ref:
|
|||
|
|
break
|
|||
|
|
# 格式转变，BGRtoRGB
|
|||
|
|
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
|||
|
|
# 转变成Image
|
|||
|
|
frame = Image.fromarray(np.uint8(frame))
|
|||
|
|
# 进行检测
|
|||
|
|
frame = np.array(yolo.detect_image(frame))
|
|||
|
|
# RGBtoBGR满足opencv显示格式
|
|||
|
|
frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
|
|||
|
|
fps = (fps + (1. / (time.time() - t1))) / 2
|
|||
|
|
# print("fps= %.2f" % (fps))
|
|||
|
|
frame = cv2.putText(frame, "fps= %.2f" % (fps), (0, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
|
|||
|
|
|
|||
|
|
cv2.imshow("video", frame)
|
|||
|
|
c = cv2.waitKey(1) & 0xff
|
|||
|
|
# print(c)
|
|||
|
|
if c == 113:
|
|||
|
|
capture.release()
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
capture.release()
|
|||
|
|
cv2.destroyAllWindows()
|