Base

.gitignore

@@ -0,0 +1,7 @@
+flagged/
+*.pt
+*.png
+*.jpg
+*.mp4
+*.mkv
+gradio_cached_examples/

app.py

@@ -0,0 +1,68 @@
+import time
+import gradio as gr
+import cv2 as cv
+from ultralytics import YOLO
+
+
+def load_yolo_model():
+    return YOLO()
+
+
+def process_video(model, cap):
+    while cap.isOpened():
+        ret, image = cap.read()
+        if not ret:
+            break
+        start = time.perf_counter()
+        results = model(image)
+        end = time.perf_counter()
+        segments = results[0].plot()
+        cv.putText(segments, f'FPS: {int(1 // (end - start))}', (10, 30),
+                   cv.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+        cv.imshow('Image Segmentation', segments)
+        key = cv.waitKey(1)
+        if key & 0xFF == ord('q'):
+            break
+    cap.release()
+    cv.destroyAllWindows()
+
+
+def segment_video(uploaded_video):
+    model = load_yolo_model()
+    cap = cv.VideoCapture(uploaded_video.name)
+    process_video(model, cap)
+
+
+def segment_webcam():
+    model = load_yolo_model()
+    cap = cv.VideoCapture(0)
+    process_video(model, cap)
+
+
+def wrapper_for_segment_video(uploaded_video):
+    segment_video(uploaded_video)
+
+
+iface = gr.Interface(
+    fn=None,
+    inputs=[gr.inputs.Checkbox(label="Segment Video"),
+            gr.inputs.Checkbox(label="Segment Webcam"),
+            gr.inputs.File(label="Upload Video")],
+    outputs=None,
+    live=True
+)
+
+
+def update_fn(segment_video_checkbox, segment_webcam_checkbox, uploaded_video):
+    if segment_video_checkbox and uploaded_video:
+        iface.fn = wrapper_for_segment_video
+        iface.args = [uploaded_video]
+    elif segment_webcam_checkbox:
+        iface.fn = segment_webcam
+        iface.args = []
+    else:
+        iface.fn = None
+
+
+iface.update(update_fn)
+iface.launch()

requirments.txt

@@ -0,0 +1,47 @@
+# Ultralytics requirements
+# Usage: pip install -r requirements.txt
+ 
+# Base ----------------------------------------
+hydra-core>=1.2.0
+matplotlib>=3.2.2
+numpy>=1.18.5
+opencv-python>=4.1.1
+Pillow>=7.1.2
+PyYAML>=5.3.1
+requests>=2.23.0
+scipy>=1.4.1
+torch>=1.7.0
+torchvision>=0.8.1
+tqdm>=4.64.0
+ultralytics
+ 
+# Logging -------------------------------------
+tensorboard>=2.4.1
+# clearml
+# comet
+ 
+# Plotting ------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+ 
+# Export --------------------------------------
+# coremltools>=6.0  # CoreML export
+# onnx>=1.12.0  # ONNX export
+# onnx-simplifier>=0.4.1  # ONNX simplifier
+# nvidia-pyindex  # TensorRT export
+# nvidia-tensorrt  # TensorRT export
+# scikit-learn==0.19.2  # CoreML quantization
+# tensorflow>=2.4.1  # TF exports (-cpu, -aarch64, -macos)
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev  # OpenVINO export
+ 
+# Extras --------------------------------------
+ipython  # interactive notebook
+psutil  # system utilization
+thop>=0.1.1  # FLOPs computation
+# albumentations>=1.0.3
+# pycocotools>=2.0.6  # COCO mAP
+# roboflow
+ 
+# HUB -----------------------------------------
+GitPython>=3.1.24