Índice
Este es el repositorio donde está almacenado todo el código desarrollado en mi TFG.
Mi proyecto consiste en el desarrollo de una herramienta que utilice la Inteligencia Artificial para generar habitaciones automáticamente de forma tridimensional usando un proyecto en Unity. Para ello, se tuvo que desarrollar un conjunto de datos con el que posteriormente se entrenó un LoRA de Stable Diffusion. También se desarrolló una aplicación web con Flask que sea capaz de comunicarse con el modelo, generar una página web y permitir el acceso mediante APIs. Finalmente, también se desarrolló un proyecto en Unity que se conectó a la aplicación Flask.
Este repositorio se puede separar en:
- Script que genera el conjunto de datos.
- Aplicación web desarrollada en Flask.
- Proyecto en Unity.
Este proyecto no está pensado para ser instalado en múltiples dispositivos. De todos modos, si se quiere probar, a continuación se explicará como hacerlo.
Debe tener instalado y habilitado el siguiente software.
-
Flask
pip install Flask
-
OpenCV
pip install opencv-python
-
ComfyUI - Instalación
-
Unity - Descarga
-
Clonar el repo
git clone https://github.com/pablo-pg/TFG.git
-
Cree un proyecto en Unity.
-
Cree un objeto vacío e insértele el script.
-
Añada los prefabs y relaciónelos en el script, así como la URL de la API.
-
Descargue los modelos de IA a utilizar y almacénelos donde la documentación de ComfyUI lo indica.
Para el uso de este proyecto, debe tener varios servicios activados.
- Ejecute ComfyUI - Repositorio
- Cargue el workflow deseado.
- Teniendo la configuración de desarrollador activada, le aparecerá el botón "Save (API Format)". Guarde el workflow o copielo en el portapapeles.
- En la apliacación Flask, modifique
app.py
y en la variableDEFAULT_PROMPT_JSON
pegue su workflow. La versión del repositorio contiene el workflow y modelos usados para el TFG. - Ejecute la aplicación Flask con
python app.py
. - En el proyecto en Unity actualice la dirección web de la API, por defecto está un enlace de prueba.
- Estando ejecutándose ComfyUI y la aplicación Flask, ejecute el proyecto en Unity.
Creado bajo MIT License. Ver LICENSE.txt
para más información.
Pablo Pérez González - alu0101318318@ull.edu.es
Enlace del repo: https://github.com/pablo-pg/TFG