Upload folder using huggingface_hub

README.md

@@ -1,5 +1,5 @@
 ---
-title: Aegis Air
+title: aegis_air
 emoji: 🐠
 colorFrom: pink
 colorTo: purple

app.py

@@ -0,0 +1,84 @@
+import numpy as np
+import supervision as sv
+from ultralytics import YOLO
+import gradio as gr
+import torch
+import spaces
+import os
+
+tracker = sv.ByteTrack()
+box_annotator = sv.BoxAnnotator()
+label_annotator = sv.LabelAnnotator()
+trace_annotator = sv.TraceAnnotator()
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model = YOLO("yolov8n.pt").to(device)
+
+FRAME_INTERVAL = 35
+LOW_RES_SIZE = (640,640)
+MID_RES_SIZE = (1280,1280)
+HIGH_RES_SIZE =(1088,1920)
+MAX_PERSONS = 0
+
+auth_users = [(os.getenv('USERNAME'), os.getenv('PASSWORD'))]
+
+@spaces.GPU
+def process_video(frame: np.ndarray, _: int) -> np.ndarray:
+    global MAX_PERSONS
+    
+    results = model(frame, imgsz=HIGH_RES_SIZE)[0]
+    detections = sv.Detections.from_ultralytics(results)
+    person_detections = detections[detections.class_id == 0]
+    
+    current_person_count = len(person_detections)
+    if current_person_count > MAX_PERSONS:
+        MAX_PERSONS = current_person_count
+    
+    print(f'Personas detectadas: {current_person_count}, Máximo: {MAX_PERSONS}')
+    
+    tracked_persons = tracker.update_with_detections(person_detections)
+    
+    labels = [
+        f"#{tracker_id} {results.names[class_id]}"
+        for class_id, tracker_id
+        in zip(tracked_persons.class_id, tracked_persons.tracker_id)
+    ]
+    
+    annotated_frame = box_annotator.annotate(frame.copy(), detections=tracked_persons)
+    annotated_frame = label_annotator.annotate(annotated_frame, detections=tracked_persons, labels=labels)
+    
+    return trace_annotator.annotate(annotated_frame, detections=tracked_persons)
+
+def upload_video(video_path):
+    global MAX_PERSONS
+    MAX_PERSONS = 0  # Reiniciar el contador para cada video
+
+    output_video_path = "output_video.mp4"
+
+    if os.path.exists(output_video_path):
+        os.remove(output_video_path)
+
+    sv.process_video(source_path=video_path, target_path=output_video_path, callback=process_video)
+    
+    return output_video_path, MAX_PERSONS
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Aegis Air Demo")
+    with gr.Row():
+        video_input = gr.Video(label="Sube tu video aquí")
+        
+            
+        video_output = gr.Video(label="Video Anotado")
+        
+    persons_output = gr.Textbox(label="Total de Personas Detectadas", interactive=False)    
+    def process_video_gradio(video):
+        if video is None:
+            return None, "No se ha subido ningún video"
+            
+        processed_video, total_persons = upload_video(video)
+        return processed_video, f"Máximo número de personas detectadas: {total_persons}"
+        
+    submit_button = gr.Button("Procesar")
+    submit_button.click(process_video_gradio, inputs=video_input, outputs=[video_output, persons_output])
+    
+demo.launch(auth=auth_users)

requirements.txt

@@ -0,0 +1,86 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.4.0
+certifi==2024.8.30
+charset-normalizer==3.3.2
+click==8.1.7
+contourpy==1.3.0
+cycler==0.12.1
+defusedxml==0.7.1
+fastapi==0.114.1
+ffmpy==0.4.0
+filelock==3.16.0
+fonttools==4.53.1
+fsspec==2024.9.0
+gradio==4.44.0
+gradio_client==1.3.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.2
+huggingface-hub==0.24.7
+idna==3.8
+importlib_resources==6.4.5
+Jinja2==3.1.4
+kiwisolver==1.4.7
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.9.2
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.3
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.6.68
+nvidia-nvtx-cu12==12.1.105
+opencv-python==4.10.0.84
+opencv-python-headless==4.10.0.84
+orjson==3.10.7
+packaging==24.1
+pandas==2.2.2
+pillow==10.4.0
+psutil==6.0.0
+py-cpuinfo==9.0.0
+pydantic==2.9.1
+pydantic_core==2.23.3
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.1.4
+python-dateutil==2.9.0.post0
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.2
+requests==2.32.3
+rich==13.8.1
+ruff==0.6.4
+scipy==1.14.1
+seaborn==0.13.2
+semantic-version==2.10.0
+setuptools==74.1.2
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.38.5
+supervision==0.23.0
+sympy==1.13.2
+tomlkit==0.12.0
+torch
+torchvision
+tqdm==4.66.5
+triton==3.0.0
+typer==0.12.5
+typing_extensions==4.12.2
+tzdata==2024.1
+ultralytics==8.2.90
+ultralytics-thop==2.0.6
+urllib3==2.2.2
+uvicorn==0.30.6
+websockets==12.0

yolov8n.pt

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