Ultralytics YOLOv5 아키텍처

YOLOv5 (v6.0/6.1) is a powerful object detection algorithm developed by Ultralytics. This article dives deep into the YOLOv5 architecture, data augmentation strategies, training methodologies, and loss computation techniques. This comprehensive understanding will help improve your practical application of object detection in various fields, including surveillance, autonomous vehicles, and image recognition.

1. 모델 구조

YOLOv5의 아키텍처는 크게 세 부분으로 구성됩니다:

• 백본: 네트워크의 본체입니다. YOLOv5 의 경우 백본은 다음을 사용하여 설계되었습니다. New CSP-Darknet53 구조는 이전 버전에서 사용된 다크넷 아키텍처를 수정한 것입니다.
• 목: 이 부분은 백본과 헤드를 연결합니다. YOLOv5 에서 , SPPF 그리고 New CSP-PAN 구조가 활용됩니다.
• Head: 이 부분은 최종 출력을 생성하는 역할을 합니다. YOLOv5 에서는 YOLOv3 Head 를 사용하세요.

모델의 구조는 아래 이미지에 나와 있습니다. 모델 구조에 대한 자세한 내용은 다음에서 확인할 수 있습니다. yolov5l.yaml.

YOLOv5 는 이전 버전과 비교하여 몇 가지 사소한 변경 사항이 있습니다:

1. 그리고 Focus 구조가 이전 버전에서 발견되는 6x6 Conv2d 구조로 변경합니다. 이 변경으로 효율성 향상 #4825.
2. 그리고 SPP 구조가 SPPF. 이 변경으로 처리 속도가 두 배 이상 빨라집니다.

속도를 테스트하려면 SPP 그리고 SPPF를 사용하면 다음 코드를 사용할 수 있습니다:

import time

import torch
import torch.nn as nn


class SPP(nn.Module):
    def __init__(self):
        """Initializes an SPP module with three different sizes of max pooling layers."""
        super().__init__()
        self.maxpool1 = nn.MaxPool2d(5, 1, padding=2)
        self.maxpool2 = nn.MaxPool2d(9, 1, padding=4)
        self.maxpool3 = nn.MaxPool2d(13, 1, padding=6)

    def forward(self, x):
        """Applies three max pooling layers on input `x` and concatenates results along channel dimension."""
        o1 = self.maxpool1(x)
        o2 = self.maxpool2(x)
        o3 = self.maxpool3(x)
        return torch.cat([x, o1, o2, o3], dim=1)


class SPPF(nn.Module):
    def __init__(self):
        """Initializes an SPPF module with a specific configuration of MaxPool2d layer."""
        super().__init__()
        self.maxpool = nn.MaxPool2d(5, 1, padding=2)

    def forward(self, x):
        """Applies sequential max pooling and concatenates results with input tensor."""
        o1 = self.maxpool(x)
        o2 = self.maxpool(o1)
        o3 = self.maxpool(o2)
        return torch.cat([x, o1, o2, o3], dim=1)


def main():
    """Compares outputs and performance of SPP and SPPF on a random tensor (8, 32, 16, 16)."""
    input_tensor = torch.rand(8, 32, 16, 16)
    spp = SPP()
    sppf = SPPF()
    output1 = spp(input_tensor)
    output2 = sppf(input_tensor)

    print(torch.equal(output1, output2))

    t_start = time.time()
    for _ in range(100):
        spp(input_tensor)
    print(f"SPP time: {time.time() - t_start}")

    t_start = time.time()
    for _ in range(100):
        sppf(input_tensor)
    print(f"SPPF time: {time.time() - t_start}")


if __name__ == "__main__":
    main()

True
SPP time: 0.5373051166534424
SPPF time: 0.20780706405639648

2. 데이터 증강 기술

YOLOv5 employs various data augmentation techniques to improve the model's ability to generalize and reduce overfitting. These techniques include:

• Mosaic Augmentation: An image processing technique that combines four training images into one in ways that encourage object detection models to better handle various object scales and translations.

  

• 복사-붙여넣기 증강: 이미지에서 무작위로 패치를 복사하여 무작위로 선택한 다른 이미지에 붙여넣어 새로운 학습 샘플을 효과적으로 생성하는 혁신적인 데이터 증강 방법입니다.

  

• 랜덤 아핀 변환: 여기에는 이미지의 무작위 회전, 크기 조정, 이동 및 전단이 포함됩니다.

  

• 믹스업 증강: 두 이미지와 관련 레이블을 선형적으로 조합하여 합성 이미지를 만드는 방법입니다.

  

• 문서화: 다양한 증강 기술을 지원하는 강력한 이미지 증강 라이브러리입니다.

• HSV 증강: 이미지의 색조, 채도 및 값을 임의로 변경합니다.

  

• 무작위 수평 뒤집기: 이미지를 무작위로 가로로 뒤집는 증강 방식입니다.

  

3. 교육 전략

YOLOv5 는 모델의 성능을 향상시키기 위해 몇 가지 정교한 훈련 전략을 적용합니다. 여기에는 다음이 포함됩니다:

• 멀티스케일 트레이닝: 훈련 과정에서 입력 이미지의 크기가 원래 크기의 0.5~1.5배 범위 내에서 임의로 조정됩니다.
• 자동 앵커: 이 전략은 사용자 지정 데이터의 기준 데이터 상자의 통계적 특성과 일치하도록 이전 앵커 상자를 최적화합니다.
• Warmup and Cosine LR Scheduler: A method to adjust the learning rate to enhance model performance.
• 지수이동평균(EMA): 과거 단계의 매개변수 평균을 사용하여 학습 과정을 안정화하고 일반화 오류를 줄이는 전략입니다.
• Mixed Precision Training: A method to perform operations in half-precision format, reducing memory usage and enhancing computational speed.
• 하이퍼파라미터 진화: 최적의 성능을 달성하기 위해 하이퍼파라미터를 자동으로 조정하는 전략입니다.

4. 추가 기능

4.1 손실 계산

YOLOv5 의 손실은 세 가지 개별 손실 구성 요소의 조합으로 계산됩니다:

• 클래스 손실(BCE 손실): 이진 교차 엔트로피 손실로, 분류 작업의 오류를 측정합니다.
• 객체성 손실(BCE 손실): 또 다른 이진 교차 엔트로피 손실로, 특정 그리드 셀에 객체가 있는지 여부를 감지할 때 발생하는 오류를 계산합니다.
• 위치 손실(CIoU 손실): 전체 IoU 손실: 그리드 셀 내에서 오브젝트의 위치를 파악할 때 발생하는 오류를 측정합니다.

The overall loss function is depicted by:

4.2 잔액 손실

세 가지 예측 레이어의 오브젝트니스 손실(P3, P4, P5)의 가중치는 다르게 적용됩니다. 밸런스 가중치는 다음과 같습니다. [4.0, 1.0, 0.4] 로 각각 설정합니다. 이 접근 방식은 서로 다른 규모의 예측이 총 손실에 적절하게 기여하도록 보장합니다.

4.3 그리드 감도 제거하기

YOLOv5 아키텍처는 이전 버전( YOLO)과 비교하여 박스 예측 전략에 몇 가지 중요한 변경 사항을 적용했습니다. YOLOv2 및 YOLOv3에서는 마지막 레이어의 활성화를 사용하여 박스 좌표를 직접 예측했습니다.

그러나 YOLOv5 에서는 그리드 감도를 줄이고 모델이 무한한 상자 치수를 예측하지 못하도록 상자 좌표를 예측하는 공식이 업데이트되었습니다.

The revised formulas for calculating the predicted bounding box are as follows:

스케일링 전후의 중심점 오프셋을 비교합니다. 중심점 오프셋 범위는 (0, 1)에서 (-0.5, 1.5)까지 조정됩니다. 따라서 오프셋은 쉽게 0 또는 1이 될 수 있습니다.

조정 전후의 높이와 너비 비율(앵커 기준)을 비교합니다. 원래 yolo/darknet 상자 방정식에는 심각한 결함이 있습니다. 폭과 높이는 단순히 아웃=엑스(인)이므로 완전히 무한대이며, 이는 폭주 기울기, 불안정성, NaN 손실 및 궁극적으로 훈련의 완전한 손실로 이어질 수 있으므로 위험합니다. 이 문제를 참조하세요.

4.4 빌드 타겟

The build target process in YOLOv5 is critical for training efficiency and model accuracy. It involves assigning ground truth boxes to the appropriate grid cells in the output map and matching them with the appropriate anchor boxes.

이 프로세스는 다음 단계를 따릅니다:

• 기준점 상자 치수와 각 앵커 템플릿의 치수의 비율을 계산합니다.

• 계산된 비율이 임계값 내에 있으면 기준값 상자를 해당 앵커와 일치시킵니다.

• 수정된 중심점 오프셋으로 인해 기준점 상자를 두 개 이상의 앵커에 할당할 수 있다는 점을 염두에 두고 일치하는 앵커를 적절한 셀에 할당합니다. 중심점 오프셋 범위가 (0, 1)에서 (-0.5, 1.5)로 조정되었기 때문입니다. GT 박스를 더 많은 앵커에 할당할 수 있습니다.

이렇게 하면 빌드 타깃 프로세스는 학습 과정에서 각 실측 객체가 적절하게 할당되고 일치하는지 확인하여 YOLOv5 객체 감지 작업을 보다 효과적으로 학습할 수 있습니다.

결론

결론적으로, YOLOv5 은 실시간 객체 감지 모델 개발의 중요한 진전을 의미합니다. 다양한 새로운 기능, 개선 사항 및 훈련 전략을 통합하여 성능과 효율성 면에서 이전 버전의 YOLO 제품군을 능가합니다.

YOLOv5 의 주요 개선 사항으로는 동적 아키텍처 사용, 광범위한 데이터 증강 기술, 혁신적인 훈련 전략, 컴퓨팅 손실 및 목표 구축 프로세스의 중요한 조정 등이 있습니다. 이러한 모든 혁신은 YOLO 모델의 트레이드마크인 빠른 속도를 유지하면서 물체 감지의 정확성과 효율성을 크게 향상시킵니다.

📅 Created 11 months ago ✏️ Updated 20 days ago

댓글