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.

Eres un traductor de españìngitaliano+djkhfs escritorystgio de documentos de TI. Traduce lo siguiente al español. Convierte las etiquetas desconocidas a Yolov5.

Tengo un conjunto de datos de imágenes con un formato de etiqueta desconocido, que es:

angry_actor_104.jpg 0 28 113 226 141 22.9362 0

Indica una imagen de la siguiente manera:

nombre_de_la_imagen id_del_rostro_en_la_imagen posición_superior_de_la_caja_del_rostro posición_izquierda_de_la_caja_del_rostro posición_derecha_de_la_caja_del_rostro posición_inferior_de_la_caja_del_rostro confianza_en_la_caja_del_rostro etiqueta_de_expresión

Mi pregunta es: ¿Cómo se puede convertir esto al formato yolov5?

He estado buscando esto durante mucho tiempo y espero que alguien pueda ayudar.
Muchas gracias de antemano.

Tags:  , ,

Answer

  1. Avatar for davy.ai

    Para convertir el conjunto de datos de imágenes con un formato de etiqueta desconocido al formato YOLOv5, puedes seguir estos pasos:

    1. Crea un archivo de texto nuevo para cada imagen en el conjunto de datos con el mismo nombre que el archivo de imagen y la extensión “.txt”. Por ejemplo, si el archivo de imagen es “angry_actor_104.jpg”, el archivo de etiquetas debería ser “angry_actor_104.txt”.
    2. Convierte las coordenadas del recuadro facial del formato actual (arriba, izquierda, derecha, abajo) al formato YOLOv5 (centro_x, centro_y, ancho, alto). Para hacer esto, puedes usar las siguientes ecuaciones:

      centro_x = (recuadro_izquierdo + recuadro_derecho) / 2 / ancho_imagen
      centro_y = (recuadro_superior + recuadro_inferior) / 2 / alto_imagen
      ancho = (recuadro_derecho - recuadro_izquierdo) / ancho_imagen
      alto = (recuadro_inferior - recuadro_superior) / alto_imagen
      

      Ten en cuenta que ancho_imagen y alto_imagen son el ancho y alto de la imagen en píxeles.

    3. Convierte la etiqueta de expresión a un índice entero que representa la clase del objeto en el formato YOLOv5. Puedes crear una correspondencia entre las etiquetas actuales y las etiquetas YOLOv5.

    4. Escribe la información de etiqueta convertida en el archivo de etiquetas en el formato YOLOv5. El formato de cada línea en el archivo de etiquetas es:

      índice_clase centro_x centro_y ancho alto
      

      Donde índice_clase es el índice entero de la clase, y el resto de los valores son las coordenadas del recuadro convertidas.

    5. Repite este proceso para todas las imágenes en el conjunto de datos. Una vez que hayas convertido todas las etiquetas, puedes entrenar un modelo YOLOv5 con el conjunto de datos en el nuevo formato.

Comments are closed.