SK AI Data Academy 1기 중급/[SKT] 통신탑 안전이상 감지 모델 개발

04. 객체 검출

Donghun Kang 2024. 9. 7. 16:48
  • R-CNN (Region Proposal CNN)

  • YOLO (You Only Look Once)

  • YOLO v1 구조
  • 총 24개의 convolution layer와 2개의 fully connected layer로 구성
  • 앞의 20개의 convolution layer는 ImageNet 데이터셋으로 pretrain
  • 뒤의 4개 convolution layer와 2개의 fully connected layer를 더 붙여서 Pascal VOC데이터로 Fine tuning

  • labelme를 이용한 데이터 라벨링