add SRC beep detection

app.py

@@ -219,8 +219,9 @@ def count_phases(phase_sin, phase_cos, threshold=0.5):
 
 def inference(in_video, use_60fps, 
               beep_detection_on, event_length,
+              miss_threshold, marks_threshold,
               count_only_api, api_key, seq_len=64, stride_length=32, stride_pad=3, batch_size=2,
-              miss_threshold=0.5, marks_threshold=0.5, median_pred_filter=True, both_feet=True, 
+              median_pred_filter=True, both_feet=True, 
               api_call=False,
               progress=gr.Progress()):
     print(in_video)
@@ -504,14 +505,14 @@ def inference(in_video, use_60fps,
 
     if LOCAL:
         if both_feet:
-            count_msg = f"## Reps Count (both feet): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
+            count_msg = f"## Count (both feet): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
         else:
-            count_msg = f"## Reps Count (one foot): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
+            count_msg = f"## Count (one foot): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
     else:
         if both_feet:
-            count_msg = f"## Reps Count (both feet): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
+            count_msg = f"## Count (both feet): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
         else:
-            count_msg = f"## Reps Count (one foot): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
+            count_msg = f"## Count (one foot): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
 
     if api_call:
         if CACHE_API_CALLS:
@@ -761,21 +762,23 @@ with gr.Blocks() as demo:
             gr.Markdown(
                 """
                 ### Inference Options
-                Select the framerate for inference.
+                Select the framerate and thresholds for inference. Default values should work well for most videos.
                 """,
                 elem_id='inference-options',
             )
             use_60fps = gr.Checkbox(label="Use 60 FPS", elem_id='use-60fps', visible=True)
+            miss_threshold = gr.Slider(label="Periodicity Threshold", minimum=0.0, maximum=1.0, step=0.05, value=0.5, elem_id='miss-threshold')
+            marks_threshold = gr.Slider(label="Marks Threshold", minimum=0.0, maximum=1.0, step=0.05, value=0.5, elem_id='marks-threshold')
         with gr.Column():
             gr.Markdown(
                 """
                 ### Beep Detection Options
-                Must be using official IJRU timing tracks.
+                Must be using official Single Rope Contest timing tracks.
                 """,
                 elem_id='beep-detection-options',
             )
             beep_detection_on = gr.Checkbox(label="Detect Beeps", elem_id='detect-beeps', visible=True)
-            event_length = gr.Textbox(label="Expected Event Length (s)", elem_id='event-length', visible=True)
+            event_length = gr.Textbox(label="Expected Event Length (s)", elem_id='event-length', visible=True, value="30")
             
     with gr.Row():
         run_button = gr.Button(value="Run", elem_id='run-button', scale=1)
@@ -803,16 +806,16 @@ with gr.Blocks() as demo:
 
     demo_inference = partial(inference, count_only_api=False, api_key=None)
     
-    run_button.click(demo_inference, [in_video, use_60fps, beep_detection_on, event_length], 
+    run_button.click(demo_inference, [in_video, use_60fps, beep_detection_on, event_length, miss_threshold, marks_threshold], 
                      outputs=[out_text, out_plot, out_phase_spiral, out_phase, out_hist])
     api_inference = partial(inference, api_call=True)
     api_dummy_button.click(api_inference, [in_video, use_60fps, beep_detection_on, event_length, count_only, api_token], 
                            outputs=[period_length], api_name='inference')
     examples = [
-        ['files/wc2023.mp4', True, True, 30],
+        ['files/wc2023.mp4', True, True, 30, 0.5, 0.5],
     ]
     gr.Examples(examples, 
-                inputs=[in_video, use_60fps, beep_detection_on, event_length], 
+                inputs=[in_video, use_60fps, beep_detection_on, event_length, miss_threshold, marks_threshold], 
                 outputs=[out_text, out_plot, out_phase_spiral, out_phase, out_hist],
                 fn=demo_inference, cache_examples=False)