40 lines
871 B
Python
40 lines
871 B
Python
import sys
|
|
import time
|
|
from multiprocessing import freeze_support
|
|
|
|
from MCamera.mn_camera import *
|
|
|
|
if __name__ == '__main__':
|
|
freeze_support()
|
|
detector = MoveNetAlgorithmPlugin()
|
|
|
|
Camera.add_source(0)
|
|
cap = Camera()
|
|
cap.start_record()
|
|
|
|
while True:
|
|
start_time = time.time()
|
|
|
|
# frame = cap.get_frame()
|
|
|
|
data = cap.get_cache()
|
|
if not data:
|
|
continue
|
|
frame = data[FRAME_DAT]
|
|
catch_time = data[CATCH_TIME]
|
|
lm_list = data[MN_RESULT]
|
|
# print(lm_list)
|
|
|
|
cv2.imshow("frame", frame)
|
|
cv2.imshow("pose_frame", detector.find_pose_with_drawing(frame))
|
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
|
break
|
|
|
|
print(time.time() - catch_time)
|
|
|
|
# print(time.time() - start_time)
|
|
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
sys.exit()
|