def pullUpAnalysis(folder_path):
'''Open folder and process .json into a dict
input: The file path
ouput: The times of push-up
'''
files = os.listdir(folder_path)
results = []
tendency = []
result = {}
r_elbow_angle_list = []
l_elbow_angle_list = []
eye_distance_list = []
tick = []
cnt = 0
for i, file in enumerate(files):
coor = jp.getJson(folder_path + '\\' + file)
if not coor:
continue
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
l_elbow_angle = getValue.getElbowAngle(coor, 'L')
eye_distance = getValue.getEyeWristDistance(coor)
if l_elbow_angle:
l_elbow_angle_list.append(l_elbow_angle)
if eye_distance:
eye_distance_list.append(eye_distance)
if r_elbow_angle:
r_elbow_angle_list.append(r_elbow_angle)
tick.append(r_elbow_angle)
if len(tick) == 5:
tend = analysis.getTendency(tick, 8) # One tick
tick = []
if tend:
tendency.append(tend)
if 3 <= len(tendency):
if tendency[-1] == 'down':
if tendency[
-2] == 'upper': # a period and tendency[-3] == 'upper'
cnt += 1
result['Num'] = cnt
standard = analysis.pullUpPeriodJudge(
r_elbow_angle_list, l_elbow_angle_list,
eye_distance_list)
result['IsRElbowStandard'], result[
'IsLElbowStandard'], result[
'IsHeightStandard'] = standard
result['Flag'] = i
r_elbow_angle_list = l_elbow_angle_list = eye_distance_list = []
results.append(result)
result = {}
return results
def pushUpAnalysis(folder_path):
'''Open folder and process .json into a dict
input: The file path
ouput: The times of push-up
Judge by tendency!
'''
files = os.listdir(folder_path)
results = []
tendency = []
result = {}
r_elbow_angle_list = []
r_knee_angle_list = []
hip_angle_list = []
# l_elbow_angle_list = []
tick = []
cnt = 0
for i, file in enumerate(files):
coor = jp.getJson(folder_path + '\\' + file)
if not coor:
continue
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
r_knee_angle = getValue.getKneeAngle(coor, 'R')
hip_angle = getValue.getHipAngle(coor, 'R')
if hip_angle:
hip_angle_list.append(hip_angle)
if r_knee_angle:
r_knee_angle_list.append(r_knee_angle)
if r_elbow_angle:
r_elbow_angle_list.append(r_elbow_angle)
tick.append(r_elbow_angle)
if len(tick) == 5:
tend = analysis.getTendency(tick, 20) # One tick
tick = []
if tend:
tendency.append(tend)
if 3 <= len(tendency):
if tendency[-1] == 'down' or tendency[-1] == 'stable':
if tendency[-2] == 'upper' and tendency[
-3] == 'upper': # a period
cnt += 1
result['Num'] = cnt
standard = analysis.pushUpPeriodJudge(
r_elbow_angle_list, hip_angle_list,
r_knee_angle_list)
result['IsRElbowStandard'], result[
'IsHipAngleStandard'], result[
'IsRKneeStandard'] = standard
result['Flag'] = i
r_elbow_angle_list = r_knee_angle_list = hip_angle_list = []
results.append(result)
result = {}
# print(tendency)
return results
def getPushUpInfos(folder_path, is_print, region=[]):
'''Open folder and process .json into a dict
input: The file path
ouput: The analysis of some infos of the whole process
'''
files = os.listdir(folder_path)
max_r_elbow_angle = 0
min_r_elbow_angle = 200
max_knee_angle = 0
min_knee_angle = 200
max_hip_angle = 0
min_hip_angle = 10000
max_hip_distance = 0
min_hip_distance = 10000
cnt = 0
cnt1 = total_elbow_angle = cnt2 = total_hip_distance = 0
cnt3 = total_knee_angle = cnt4 = total_hip_angle = 0
elb_ang = []
hip_dis = []
knee_ang = []
hip_ang = []
for i, file in enumerate(files):
coor = jp.getJson(folder_path + '\\' + file)
if not coor:
continue
cnt = i
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
r_knee_angle = getValue.getKneeAngle(coor, 'R')
hip_angle = getValue.getHipAngle(coor, 'R')
hip_distance = getValue.getHipDistance(coor, 'R')
if r_elbow_angle:
elb_ang.append(r_elbow_angle)
total_elbow_angle += r_elbow_angle
cnt1 += 1
max_r_elbow_angle = max(max_r_elbow_angle, r_elbow_angle)
min_r_elbow_angle = min(min_r_elbow_angle, r_elbow_angle)
else:
elb_ang.append(0)
if hip_distance:
hip_dis.append(hip_distance)
total_hip_distance += hip_distance
cnt2 += 1
max_hip_distance = max(max_hip_distance, hip_distance)
min_hip_distance = min(min_hip_distance, hip_distance)
else:
hip_dis.append(0)
if r_knee_angle:
knee_ang.append(r_knee_angle)
total_knee_angle += r_knee_angle
cnt3 += 1
max_knee_angle = max(max_knee_angle, r_knee_angle)
min_knee_angle = min(min_knee_angle, r_knee_angle)
else:
knee_ang.append(0)
if hip_angle:
hip_ang.append(hip_angle)
total_hip_angle += hip_angle
cnt4 += 1
max_hip_angle = max(max_hip_angle, hip_angle)
min_hip_angle = min(min_hip_angle, hip_angle)
else:
hip_ang.append(0)
aver_r_elbow_angle = total_elbow_angle / cnt1
aver_hip_distance = total_hip_distance / cnt2
aver_knee_angle = total_knee_angle / cnt3
aver_hip_angle = total_hip_angle / cnt4
if is_print:
print('max elbow angle:', max_r_elbow_angle)
print('min elbow angle:', min_r_elbow_angle)
print('aver elbow angle:', aver_r_elbow_angle)
print('max hip distance:', max_hip_distance)
print('min hip distance:', min_hip_distance)
print('aver hip distance:', aver_hip_distance)
print('max knee angle:', max_knee_angle)
print('min knee angle:', min_knee_angle)
print('aver knee angle:', aver_knee_angle)
print('max hip angle:', max_hip_angle)
print('min hip angle:', min_hip_angle)
print('aver hip angle:', aver_hip_angle)
elb_ang_below_105 = []
for ans in elb_ang:
if ans < 105:
elb_ang_below_105.append(ans)
else:
elb_ang_below_105.append(None)
cnt += 1
ax1 = plt.subplot(2, 2, 1)
# plt.hlines(165, )
# ax1.scatter(range(cnt1), elb_ang, label='elbow angle', s=10)
# ax1.scatter(range(cnt1), elb_ang_below_105, label='elbow angle', color='r', s=10)
# actu = [89, 149, 238, 328, 418, 597, 686, 746, 836, 955, 985]
# region = 0
ax1.scatter(range(cnt), elb_ang, label='elbow angle', s=2)
ax1.scatter(range(cnt), elb_ang_below_105, label='elbow angle', color='r', s=2)
ax1.hlines(165, 0, cnt, colors='c')
ax1.vlines(region, 0, 180)
# ax1.vlines(actu, 0, 180, colors='r')
ax1.set_title('elbow_angle')
ax2 = plt.subplot(2, 2, 2)
ax2.scatter(range(cnt), hip_dis, label='hip distance', s=2)
ax2.set_title('hip_distance')
ax3 = plt.subplot(2, 2, 3)
ax3.scatter(range(cnt), knee_ang, label='knee angle', s=2)
ax3.vlines(region, 0, 180)
# ax3.vlines(actu, 0, 180, colors='r')
ax3.set_title('knee_angle')
ax4 = plt.subplot(2, 2, 4)
ax4.scatter(range(cnt), hip_ang, label='hip', s=2)
ax4.vlines(region, 0, 180)
# ax4.vlines(actu, 0, 180, colors='r')
ax4.set_title('hip_angle')
plt.show()
def getPullUpInfos(folder_path, is_print, region=[]):
'''Open folder and process .json into a dict
input: The file path
ouput: The analysis of some infos of the whole process
'''
files = os.listdir(folder_path)
max_r_elbow_angle = max_l_elbow_angle = 0
min_r_elbow_angle = min_l_elbow_angle = 200
r_elb_ang = []
l_elb_ang = []
neck_horizontalbar_distance = []
eye_wrist_distance_list = []
cnt1 = cnt2 = 0
total_r_elbow_angle = total_l_elbow_angle = 0
for i, file in enumerate(files):
coor = jp.getJson(folder_path + '\\' + file)
if not coor:
continue
cnt = i
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
l_elbow_angle = getValue.getElbowAngle(coor, 'L')
head_wrist_distance = getValue.getHeadDistance(coor)
eye_wrist_distance = getValue.getEyeWristDistance(coor)
if r_elbow_angle:
r_elb_ang.append(r_elbow_angle)
total_r_elbow_angle += r_elbow_angle
cnt1 += 1
max_r_elbow_angle = max(max_r_elbow_angle, r_elbow_angle)
min_r_elbow_angle = min(min_r_elbow_angle, r_elbow_angle)
else:
r_elb_ang.append(0)
if l_elbow_angle:
l_elb_ang.append(l_elbow_angle)
total_l_elbow_angle += l_elbow_angle
cnt2 += 1
max_l_elbow_angle = max(max_l_elbow_angle, l_elbow_angle)
min_l_elbow_angle = min(min_l_elbow_angle, l_elbow_angle)
else:
l_elb_ang.append(0)
if head_wrist_distance:
neck_horizontalbar_distance.append(head_wrist_distance)
else:
neck_horizontalbar_distance.append(0)
if eye_wrist_distance:
eye_wrist_distance_list.append(eye_wrist_distance)
else:
eye_wrist_distance_list.append(0)
aver_r_elbow_angle = total_r_elbow_angle / cnt1
aver_l_elbow_angle = total_l_elbow_angle / cnt2
if is_print:
print('max r_elbow angle:', max_r_elbow_angle)
print('min r_elbow angle:', min_r_elbow_angle)
print('aver r_elbow angle:', aver_r_elbow_angle)
print('max l_elbow angle:', max_l_elbow_angle)
print('min l_elbow angle:', min_l_elbow_angle)
print('aver l_elbow angle:', aver_l_elbow_angle)
# elb_ang_below_105 = []
# for ans in elb_ang:
# if ans < 105:
# elb_ang_below_105.append(ans)
# else:
# elb_ang_below_105.append(None)
cnt += 1
ax1 = plt.subplot(2, 2, 1)
ax1.scatter(range(cnt), r_elb_ang, label='relbow angle', s=2)
ax1.vlines(region, 0, 180)
ax1.set_title('relbow_angle')
ax2 = plt.subplot(2, 2, 2)
ax2.scatter(range(cnt), l_elb_ang, label='lelbow angle', s=2)
ax2.set_title('lelbow_angle')
ax3 = plt.subplot(2, 2, 3)
ax3.scatter(range(cnt), neck_horizontalbar_distance, label='neck bar distance', s=2)
ax3.set_title('neck bar distance')
ax4 = plt.subplot(2, 2, 4)
ax4.scatter(range(cnt), eye_wrist_distance_list, label='eye bar distance', s=2)
ax4.set_title('eye bar distance')
plt.show()
def pullUpDetect(length):
# Process
cnt = 0
results = []
tendency = []
result = {}
r_elbow_angle_list = []
l_elbow_angle_list = []
eye_distance_list = []
tick = []
pullUpCnt = 0
# Starting OpenPose
opWrapper = op.WrapperPython()
opWrapper.configure(params)
opWrapper.start()
# get video from webcam or video
start = time.time()
cap = cv2.VideoCapture(0)
cap_t = cv2.VideoCapture(1)
# cap2 = cv2.VideoCapture(r'E:\University\科研创新\雏燕计划-体测\体测姿势素材\push-up\push-up-test-1.mp4')
coorList = []
start_time = datetime.now()
while True:
# Get images from cam
ret, imageToProcess = cap.read()
ret, imageToTest = cap_t.read()
cv2.imshow('video', imageToProcess)
cv2.imshow('2', imageToTest)
cur_time = datetime.now()
delta_time = (cur_time - start_time).seconds
print('\r倒计时：' + length - delta_time + '秒', end='')
if delta_time == length:
break
if cnt % 2 == 0:
datum = op.Datum()
datum.cvInputData = imageToProcess
opWrapper.emplaceAndPop([datum])
if datum.poseKeypoints.size != 1:
coor = kpp.getKeyPoints(datum.poseKeypoints[0])
if coor:
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
l_elbow_angle = getValue.getElbowAngle(coor, 'L')
eye_distance = getValue.getEyeWristDistance(coor)
if l_elbow_angle:
l_elbow_angle_list.append(l_elbow_angle)
if eye_distance:
eye_distance_list.append(eye_distance)
if r_elbow_angle:
r_elbow_angle_list.append(r_elbow_angle)
tick.append(r_elbow_angle)
if len(tick) == 5:
tend = analysis.getTendency(tick, 8) # One tick
tick = []
if tend:
tendency.append(tend)
if 3 <= len(tendency):
if tendency[-1] == 'down' or tendency[
-1] == 'stable':
if tendency[-2] == 'upper' and tendency[
-3] == 'upper': # a period
pullUpCnt += 1
result['Num'] = pullUpCnt
standard = analysis.pullUpPeriodJudge(
r_elbow_angle_list,
l_elbow_angle_list,
eye_distance_list)
result['IsRElbowStandard'], result[
'IsLElbowStandard'], result[
'IsHeightStandard'], total = standard
r_elbow_angle_list = l_elbow_angle_list = eye_distance_list = []
results.append(result)
print(result)
result = {}
db.ret(delta_time, total, pullUpCnt,
'pullUp', results, True)
cv2.imshow("OpenPose 1.5.1 - Tutorial Python API",
datum.cvOutputData)
if cv2.waitKey(1) == ord('q'):
break
cnt += 1
db.ret(None, None, None, None, None, False)
def pushUpDetect(length):
# Process
cnt = 0
# Starting OpenPose
opWrapper = op.WrapperPython()
opWrapper.configure(params)
opWrapper.start()
# get video from webcam or video
start = time.time()
cap_side = cv2.VideoCapture(0)
cap_front = cv2.VideoCapture(1)
# cap2 = cv2.VideoCapture(r'E:\University\科研创新\雏燕计划-体测\体测姿势素材\push-up\push-up-test-1.mp4')
coorList = []
results = []
tendency = []
result = {}
r_elbow_angle_list = []
l_elbow_angle_list = []
l_knee_angle_list = []
hip_angle_list = []
hip_distance_list = []
tick = []
pushUpCnt = 0
start_time = datetime.now()
while True:
# Get images from cam
ret, imageToProcessFront = cap_front.read()
ret, imageToProcessSide = cap_side.read()
cv2.imshow('front', imageToProcessFront)
cv2.imshow('side', imageToProcessSide)
cnt = 0
cur_time = datetime.now()
delta_time = (cur_time - start_time).seconds
print('\r倒计时：' + length - delta_time + '秒', end='')
if delta_time == length:
break
if cnt % 2 == 0:
datum_front = op.Datum()
datum_side = op.Datum()
datum_front.cvInputData = imageToProcessFront
datum_side.cvInputData = imageToProcessSide
opWrapper.emplaceAndPop([datum_front])
opWrapper.emplaceAndPop([datum_side])
# print("Body keypoints:")
if datum_front.poseKeypoints.size != 1 and datum_side.poseKeypoints.size != 1:
coor_front = kpp.getKeyPoints(
datum_front.poseKeypoints[0]) # 记得改参数
coor_side = kpp.getKeyPoints(datum_side.poseKeypoints[0])
r_elbow_angle = getValue.getElbowAngle(coor_front, 'R')
l_elbow_angle = getValue.getElbowAngle(coor_front, 'L')
l_knee_angle = getValue.getKneeAngle(coor_side, 'L')
hip_angle = getValue.getHipAngle(coor_side, 'L')
hip_distance = getValue.getHipDistance(coor_side, 'L')
if r_elbow_angle:
r_elbow_angle_list.append(r_elbow_angle)
tick.append(r_elbow_angle)
if hip_angle:
hip_angle_list.append(hip_angle)
if l_knee_angle:
l_knee_angle_list.append(l_knee_angle)
if l_elbow_angle:
l_elbow_angle_list.append(l_elbow_angle)
if hip_distance:
hip_distance_list.append(hip_distance)
if len(tick) == 5:
tend = analysis.getTendency(tick, 20) # One tick
print(tend)
tick = []
if tend:
tendency.append(tend)
if 3 <= len(tendency):
if tendency[-1] == 'down' or tendency[
-1] == 'stable':
if tendency[
-2] == 'upper': # a period and tendency[-3] == 'upper'
pushUpCnt += 1
result['Num'] = pushUpCnt
standard = analysis.pushUpPeriodJudgeTwoSides(
r_elbow_angle_list,
l_elbow_angle_list, hip_angle_list,
l_knee_angle_list,
hip_distance_list)
result['IsRElbowStandard'], result[
'IsLElbowStandard'], result[
'IsHipAngleStandard'], result[
'IsRKneeStandard'], result[
'IsHipDistanceStandard'], total = standard
r_elbow_angle_list = []
l_elbow_angle_list = []
l_knee_angle_list = []
hip_angle_list = []
hip_distance_list = []
pushUpCnt += 1
results.append(result)
print(result)
result = {}
db.ret(delta_time, total, pushUpCnt,
'pushUp', results, True)
cnt += 1
cv2.imshow("front", datum_front.cvOutputData)
cv2.imshow("side", datum_side.cvOutputData)
if cv2.waitKey(1) == ord('q'):
break
db.ret(None, None, None, None, None, False)
def pushUpVisualization(coor, i):
'''Read coordinates and display it
input: List of coordinates
ouput: The analysis of some infos of the whole process
'''
plt.ion()
# for i, coor in enumerate(coorList):
# if not coor:
# continue
cnt = i
r_elbow_angle = getValue.getElbowAngle(coor, 'R')
r_knee_angle = getValue.getKneeAngle(coor, 'R')
hip_angle = getValue.getHipAngle(coor, 'R')
hip_distance = getValue.getHipDistance(coor, 'R')
# if r_elbow_angle:
# elb_ang.append(r_elbow_angle)
# else:
# elb_ang.append(0)
# if hip_distance:
# hip_dis.append(hip_distance)
# else:
# hip_dis.append(0)
# if r_knee_angle:
# knee_ang.append(r_knee_angle)
# else:
# knee_ang.append(0)
# if hip_angle:
# hip_ang.append(hip_angle)
# else:
# hip_ang.append(0)
# elb_ang_below_105 = []
# for ans in elb_ang:
# if ans < 105:
# elb_ang_below_105.append(ans)
# else:
# elb_ang_below_105.append(None)
# cnt += 1
ax1 = plt.subplot(2, 2, 1)
# plt.hlines(165, )
# ax1.scatter(range(cnt1), elb_ang, label='elbow angle', s=10)
# ax1.scatter(range(cnt1), elb_ang_below_105, label='elbow angle', color='r', s=10)
# actu = [89, 149, 238, 328, 418, 597, 686, 746, 836, 955, 985]
# region = 0
ax1.scatter(i, r_elbow_angle, label='elbow angle', s=2)
# ax1.scatter(i, elb_ang_below_105, label='elbow angle', color='r', s=2)
ax1.hlines(165, 0, cnt, colors='c')
# ax1.vlines(actu, 0, 180, colors='r')
ax1.set_title('elbow_angle')
ax2 = plt.subplot(2, 2, 2)
ax2.scatter(i, hip_distance, label='hip distance', s=2)
ax2.set_title('hip_distance')
ax3 = plt.subplot(2, 2, 3)
ax3.scatter(i, r_knee_angle, label='knee angle', s=2)
# ax3.vlines(actu, 0, 180, colors='r')
ax3.set_title('knee_angle')
ax4 = plt.subplot(2, 2, 4)
ax4.scatter(i, hip_angle, label='hip', s=2)
# ax4.vlines(actu, 0, 180, colors='r')
ax4.set_title('hip_angle')
# plt.pause(0.01)
plt.plot()