update: Add examples.

.gitattributes

@@ -11,6 +11,7 @@
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
 *.mlmodel filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
+*.mp4 filter=lfs diff=lfs merge=lfs -text
 *.msgpack filter=lfs diff=lfs merge=lfs -text
 *.npy filter=lfs diff=lfs merge=lfs -text
 *.npz filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,24 +1,14 @@
-import os
-import tempfile
 import gradio as gr
 import tensorflow as tf
 from moviepy.editor import VideoFileClip
 from ultralytics import YOLO
 
+from core.data import ClassMapping
 from core.model import load_classifier
 from core.inference import FrameProcessor
 
 print("Tensorflow version " + tf.__version__)
 
-id_to_name = {
-  0: 'Flying',
-  1: 'Landing',
-  2: 'Other',
-  3: 'Straight Taxiing',
-  4: 'Takeoff',
-  5: 'Turning Maneuver',
-}
-
 print('Load classifier.')
 classifier_path = 'weights/classifier-7.keras'
 classifier = load_classifier(classifier_path)
@@ -29,20 +19,33 @@ detector = YOLO(detector_path)
 
 def fn(video: gr.Video):
   print('Process video.')
-  with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as f:
-    output = f.name
-    clip = VideoFileClip(video)
-    process_frame = FrameProcessor(detector, classifier, id_to_name)
-    clip = clip.fl_image(process_frame)
-    clip.write_videofile(output, fps=clip.fps, audio_codec='aac', logger=None)
+  output = f'Marked-{str(video)}'
+  clip = VideoFileClip(video)
+  id_to_name = {
+    0: 'Flying',
+    1: 'Landing',
+    2: 'Other',
+    3: 'Straight Taxiing',
+    4: 'Takeoff',
+    5: 'Turning Maneuver',
+  }
+  process_frame = FrameProcessor(detector, classifier, id_to_name)
+  clip = clip.fl_image(process_frame)
+  clip.write_videofile(output, fps=clip.fps, audio_codec='aac', logger=None)
   return output
 
 inputs = gr.Video(sources=['upload'], label='Input Video')
 outputs = gr.Video(interactive=False, label='Aeroplane Position and Action Marked')
 
+examples = [
+  ['examples/ZFLFDfovqls_001310_001320.mp4'] # cspell: disable-line
+  ['examples/Zv7GyH-fpEY_2023.0_2033.0.mp4']
+]
+
 iface = gr.Interface(
   fn=fn,
   inputs=inputs,
   outputs=outputs,
+  examples=examples,
 )
 iface.launch()

examples/ZFLFDfovqls_001310_001320.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1cf18a000dac495924649008dfe29a413657e2cbd9c39400f4980867de39dba4
+size 669598

examples/Zv7GyH-fpEY_2023.0_2033.0.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dec949ee31a5e3b95f0b12a67c2abdcaf8d2e4fdab6dcae81d9cf563f77851a1
+size 1055493