¿Cómo podemos agregar una clase extra al modelo YOLOV5 existente?

Answer

1. 

   davy.ai

   July 14, 2023 at 10:17 am

   Para agregar una clase adicional a las 80 clases existentes en YOLOV5, puedes seguir estos pasos:

   1. Crea un nuevo archivo de texto llamado custom.names y añade el nombre de tu clase personalizada en él.

   2. Modifica el archivo de configuración yolov5l.yaml existente agregando el número de clases (81) al parámetro nc y la ruta al archivo custom.names al parámetro names. Debería lucir así:

   nc: 81
   names:
     - "persona"
     - "bicicleta"
     ...
     - "señal de alto"
     - "clase personalizada"
   

   1. Coloca tus imágenes personalizadas en una nueva carpeta y etiquétalas usando una herramienta como LabelImg. Guarda las etiquetas como archivos XML.

   2. Convierte las etiquetas XML al formato YOLO utilizando un script como labelImg.py.

   3. Combina el conjunto de datos existente con el conjunto de datos personalizado utilizando un script como concat_data.py.

   4. Entrena el modelo utilizando el script train.py con el nuevo archivo de configuración y conjunto de datos.

   Siguiendo estos pasos, puedes agregar tu clase personalizada a las 80 clases existentes en YOLOV5 sin perder la información preentrenada.

Comments are closed.