while True:
return_value, frame = vid.read()
#cv2.imwrite("image.jpg", frame)
#image.show()
#------------------
# image = Image.fromarray(cv2.cvtColor(frame,cv2.COLOR_BGR2RGB))
-------------------------------------------------(添加以下部分代码,判断是否空帧)
if(return_value ==False):
break
else:
----------------------------------------
image = Image.fromarray(frame)
image,lab = yolo.detect_image(image)
#______________________________________修改_______________________
#st.denoise(lab)
#----------------------------------
result = np.asarray(image)
# result = cv2.cvtColor(np.asarray(image), cv2.COLOR_RGB2BGR)
curr_time = timer()
exec_time = curr_time - prev_time
prev_time = curr_time
accum_time = accum_time + exec_time
curr_fps = curr_fps + 1
if accum_time > 1:
accum_time = accum_time - 1
fps = "FPS: " + str(curr_fps)
curr_fps = 0
cv2.putText(result, text=fps, org=(3, 15), fontFace=cv2.FONT_HERSHEY_SIMPLEX,
fontScale=0.50, color=(255, 0, 0), thickness=2)
cv2.namedWindow("result", cv2.WINDOW_NORMAL)
cv2.imshow("result", result)
if isOutput:
out.write(result)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
保存路劲切记加上文件名。