es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Las imágenes de StyleGAN3 se vuelven sub o sobreexpuestas con un conjunto de datos en evolución.

Estoy intentando utilizar StyleGAN3 con un conjunto de datos en evolución, mi lógica actual es la siguiente:

  • Entrenar durante aproximadamente media hora.
  • Regenerar el conjunto de datos con imágenes agregadas.
  • Reanudar la formación con –resume.

Para las pruebas, he estado utilizando el conjunto de datos metfaces y comencé con alrededor de 100 imágenes, agregando alrededor de 50 nuevas imágenes al conjunto de datos cada media hora.

Al reanudar la formación, los resultados gradualmente se vuelven muy oscuros o muy claros.

Estoy entrenando con los parámetros sugeridos en GitHub para StyleGAN3:

train.py --cfg=stylegan3-r --gpus=1 --batch=32 --gamma=2 --batch-gpu=8 --mirror=1

¿Alguna sugerencia sobre cómo proceder al entrenar con un conjunto de datos en crecimiento?

Tags:  , ,

Answer

  1. Avatar for davy.ai

    Utilizando StyleGAN3 con un Dataset Evolutivo

    Si estás entrenando StyleGAN3 con un dataset evolutivo, y los resultados gradualmente se vuelven muy oscuros o muy claros al reanudar el entrenamiento con el comando --resume, aquí hay algunas sugerencias para mejorar el proceso de entrenamiento:

    1. Utilice tasas de aprendizaje adaptativas: La tasa de aprendizaje de un modelo determina qué tan rápido se adapta a los nuevos datos. Si la tasa de aprendizaje es demasiado alta, el modelo puede sobrepasar y ignorar nuevos datos. Por otro lado, si la tasa de aprendizaje es demasiado baja, el modelo puede no aprender nuevos datos. Utilice tasas de aprendizaje adaptativas para ajustar la tasa de aprendizaje basada en el estado actual del modelo.
    2. Entrene por más tiempo antes de regenerar el dataset: Media hora es un tiempo corto para entrenar un modelo GAN. Por lo tanto, puede ser necesario entrenar durante más tiempo antes de regenerar el dataset. De esta forma, el modelo tendrá más datos con los que trabajar y podrá ajustarse mejor a los nuevos datos.

    3. Utilice aumento de datos: El aumento de datos es el proceso de generar nuevos datos variando los datos existentes. Puede utilizar el aumento de datos para crear variaciones en su dataset existente. Esto puede ser útil cuando se tiene un dataset pequeño y se quiere entrenar un modelo que pueda manejar un dataset más grande.

    4. Ajuste los parámetros de entrenamiento: Ajustar los parámetros de entrenamiento como el tamaño del lote, gamma y mirror puede afectar el proceso de entrenamiento. Experimente con diferentes valores para estos parámetros para encontrar los valores óptimos para su dataset.

    En general, el entrenamiento con un dataset en crecimiento puede ser desafiante, pero seguir estas sugerencias puede ayudarlo a mejorar la calidad de sus resultados.

Comments are closed.