Initial commit UNDO-CCTV detector

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from ultralytics import YOLO
+
+# Load trained YOLOv8 model
+model = YOLO('samples/best.pt')
+
+def detect_objects(image):
+    """
+    Run inference on the uploaded image.
+    :param image: Input image for object detection.
+    :return: Image with bounding boxes and labels.
+    """
+    results = model.predict(image)
+    return results[0].plot()  # Plots the bounding boxes on the image
+
+# Gradio interface
+demo = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil"),
+    outputs="image",
+    title="YOLOv8 CCTV Detection",
+    description="Upload an image to detect CCTV on public spaces using a YOLOv8 model."
+)
+
+if __name__ == "__main__":
+    demo.launch()

best.pt

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

requirements.txt

@@ -0,0 +1,16 @@
+pillow-heif==0.21.0
+pillow==11.1.0
+loguru==0.7.3
+playwright==1.47.0
+pandas==2.2.3
+torch==2.7.0
+torchvision==0.22.0
+scikit-learn==1.6.1
+ultralytics==8.3.74
+pyyaml==6.0.2
+sphinx==8.2.3
+sphinx-rtd-theme==3.0.2
+gradio==5.29.0
+
+ruff==0.9.10
+pre-commit==3.8.0