File size: 886 Bytes
97a245c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from microprograms.errors.angles_micro_programs import applyPositionTightnessError
# from models.pose_estimator.pose_estimator_model_setup import get_pose_estimation
def somersault_microprogram_one_frame(filepath, on_board, expected_som, half_som_count, expected_twists, petal_count, pose_pred=None, diver_detector=None, pose_model=None):
if on_board:
return 0
if expected_som <= half_som_count:
return 0
# if not done with som or twists, need to determine if som or twist
angle = applyPositionTightnessError(filepath, pose_pred, diver_detector=diver_detector, pose_model=pose_model)
if angle is None:
return 0
# if not done with som but done with twists
if expected_som > half_som_count and expected_twists <= petal_count:
return 1
# print("angle:", angle)
if angle <= 80:
return 1
else:
return 0 |