Commit 3de0856d authored by DeriZSY's avatar DeriZSY
Browse files

addition

parent 6c349a11
Loading
Loading
Loading
Loading
+7 −7
Original line number Diff line number Diff line
@@ -240,13 +240,13 @@ def track(mat, color_threshold, square_ratio, angle_rate, length_rate, matrix_th

    # Calculate angle matrix
    matrix_bar_angle_diff = []
    for i in range(len(bar_angle)):
    for i in range(len(bar_angle)):# bar_angle : all bar angles
        temp_one_line = []
        for j in range(len(bar_angle)):
            if j <= i:
                temp_one_line.append(0)
            else:
                temp_one_line.append(abs(bar_angle[i] - bar_angle[j]))
                temp_one_line.append(abs(bar_angle[i] - bar_angle[j])) # get relative bar angles
        matrix_bar_angle_diff.append(temp_one_line)
    matrix_bar_angle_diff = np.array(matrix_bar_angle_diff)
    if MODE == D: