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()
|