Un proyecto de visión por computador que detecta y rastrea logos de Adidas en imágenes y videos utilizando YOLOv5.
Este proyecto implementa un sistema de detección de logos basado en aprendizaje profundo, específicamente entrenado para identificar logos de Adidas tanto en imágenes estáticas como en contenido de video. El sistema utiliza la arquitectura YOLOv5 y proporciona análisis detallados sobre las apariciones del logo.
- 📷 Detección de logos en tiempo real en imágenes
- 🎥 Procesamiento de videos con análisis detallado
- 📊 Estadísticas y métricas de detección
- 💻 Interfaz de usuario amigable con Streamlit
- 🔄 Capacidades de procesamiento por lotes
- 📈 Análisis de rendimiento y visualización
logo-detection-project/
├── datasets/
│ └── OpenLogo-Dataset/
│ ├── Annotations/
│ ├── ImageSets/
│ └── JPEGImages/
├── data/
├── models/
│ └── logo_detection/
│ └── weights/
├── results/
├── src/
│ ├── app6.py # Interfaz web Streamlit
│ ├── preprocessing.py # Utilidades de preprocesamiento
│ ├── train.py # Script de entrenamiento
│ └── video_detection_8.py # Módulo de procesamiento de video
├── uploaded_videos/
└── yolov5/ # Submódulo YOLOv5
- Python 3.12.4+
- PyTorch 2.5.1+
- OpenCV
- Streamlit
- YOLOv5
- CUDA (opcional, para aceleración GPU)
- Clonar el repositorio:
git clone https://github.com/tuusuario/logo-detection-project.git
cd logo-detection-project
- Crear y activar el entorno virtual:
python -m venv myenv
source myenv/bin/activate # Linux/Mac
myenv\Scripts\activate # Windows
- Instalar dependencias:
pip install -r requirements.txt
- Clonar el repositorio YOLOv5:
git clone https://github.com/ultralytics/yolov5.git
cd yolov5
pip install -r requirements.txt
python src/train.py --epochs 100 --batch-size 16 --img-size 640
streamlit run src/app6.py
from video_detection_8 import detect_logo_in_video
resultados = detect_logo_in_video(
video_path="ruta/al/video.mp4",
weights_path="models/logo_detection/weights/best.pt",
conf_threshold=0.5
)
El sistema proporciona métricas completas de detección incluyendo:
- Tiempo total de detección
- Duración de aparición del logo
- Porcentaje de presencia del logo
- Análisis cuadro por cuadro
- Puntuaciones de confianza
- Conteo de detecciones
¡Las contribuciones son bienvenidas! No dudes en enviar un Pull Request.
- YOLOv5 por la arquitectura de detección de objetos
- OpenLogo Dataset por los datos de entrenamiento
- Streamlit por el framework de interfaz web
Para cualquier pregunta o sugerencia, por favor abre un issue en el repositorio o contacta a los mantenedores.
Javier Gregoris y Alejandra Piñango.