Proyecto de detección de cubos de Rubik en tiempo real usando inteligencia artificial (YOLOv8) y cámara web.
# Activar entorno virtual
entornoSecure\Scripts\activate
# Ejecutar detector principal
python main.py# Probar modelo entrenado
python test_webcam.pysecureSystem/
├── PRODUCCIÓN
│ ├── main.py # Detector principal con YOLO
│ ├── test_webcam.py # Prueba del modelo entrenado
│ ├── config.py # Configuración del sistema
│ ├── camera_handler.py # Manejo de cámara
│ └── requirements.txt # Dependencias Python
│
├── DESARROLLO (scripts/)
│ ├── train_model.py # Entrenar modelo YOLO
│ ├── capture_images.py # Capturar imágenes
│ ├── label_images.py # Etiquetar dataset
│ ├── review_labels.py # Revisar etiquetas
│ ├── prepare_dataset.py # Preparar dataset YOLO
│ └── test_camera.py # Encontrar índice cámara
│
├── TESTING
│ ├── README.md # Docs para Surface Pro 5
│ ├── camera_frontal.py # Pruebas Linux/libcamera
│ ├── ov5693.yaml # Config cámara frontal
│ └── ov8865.yaml # Config cámara trasera
│
├── MODELO IA
│ └── runs/detect/rubik_detector2/weights/best.pt # Modelo entrenado
│
├── DATASETS
│ ├── dataset/ # Imágenes originales
│ └── yolo_dataset/ # Dataset formato YOLO
│
└── SISTEMA
├── entornoSecure/ # Entorno virtual Python
└── .gitignore # Control de versiones
- GPU: NVIDIA RTX 3070 con CUDA
- PyTorch: 2.7.1+cu118 (optimizado para CUDA)
- Modelo: YOLOv8 nano (3M parámetros)
- Rendimiento: ~30-60 FPS en tiempo real
| Tecla | Acción |
|---|---|
Q |
Salir del programa |
R |
Resetear contador de detecciones |
- Dataset: 283 imágenes etiquetadas
- Entrenamiento: 226 imágenes (80%)
- Validación: 57 imágenes (20%)
- Época: 50 iteraciones
- Precisión: Alta detección de cubos de Rubik
Interfaz principal mostrando detección con confianza del 95%
Progreso de entrenamiento: Pérdida y precisión por época
cd scripts
python train_model.pycd scripts
python capture_images.py # Capturar fotos
python label_images.py # Etiquetar manualmente
python review_labels.py # Revisar y limpiar
python prepare_dataset.py # Preparar para YOLODavid Riquelme
Chile
david.riquelme.sb@gmail.com
