YOLOV5 ha trabajado mucho en la mejora de imágenes. La mejora de imágenes se realiza entre partes del código. Se puede decir que las funciones se están automatizando cada vez más, y el grado de participación manual es cada vez menor. Echemos un vistazo más de cerca a los métodos de mejora de imágenes utilizados por YOLOv5.
1. mosaico
Aquí yolov5 también propone una operación de mosaico de 9 imágenes. 9 imágenes se empalman juntas para su procesamiento. La imagen es más grande y hay más etiquetas. Entrenar tal imagen empalmada es equivalente a entrenar 9 imágenes pequeñas.
2. Mejora de copiar y pegar datos
Pegue aleatoriamente parte del objetivo en la imagen
3. Afín aleatorio
Realice aleatoriamente transformaciones afines, es decir, escalado y traducción.
4.Mezclar
Es mezclar dos imágenes juntas de acuerdo a una cierta transparencia.
5.Albumentaciones，
Es principalmente para hacer un poco de filtrado, ecualización de histogramas, cambiar la calidad de la imagen, etc. Veo que el código escrito en el código solo se habilitará si el paquete de albumentaciones está instalado, pero el paquete de albumentaciones está comentado en el archivo de requisitos.txt del proyecto, por lo que no está habilitado de forma predeterminada.
6.Aumento de HSV (tono, saturación, valor)
Ajusta aleatoriamente el tono, la saturación y la luminosidad.
7. Volteo horizontal aleatorio
volteo horizontal aleatorio