본문 바로가기
  • 인공지능
  • 블록체인
  • 정보보안
신기술분석/AI 인공지능

YOLOv3 객체 검출

by nathan03 2021. 8. 31.
반응형

■ YOLO란 ?

 

YOLO: Real-Time Object Detection

YOLO: Real-Time Object Detection You only look once (YOLO) is a state-of-the-art, real-time object detection system. On a Pascal Titan X it processes images at 30 FPS and has a mAP of 57.9% on COCO test-dev. Comparison to Other Detectors YOLOv3 is extremel

pjreddie.com

■ YOLOv3

 

YOLOv3: An Incremental Improvement

We present some updates to YOLO! We made a bunch of little design changes to make it better. We also trained this new network that's pretty swell. It's a little bigger than last time but more accurate. It's still fast though, don't worry. At 320x320 YOLOv3

arxiv.org

  • 기존 객체 검출 방법과 성능은 비슷하고 속도는 훨씬 빠름 
  • COCO 데이터셋 사용 (80개 클래스 객체 검출)
  • https://cocodataset.org
 

COCO - Common Objects in Context

 

cocodataset.org

  • YOLO는 속도가 50ms 이하의 좋은 성능을 보임 

  • 학습 데이터 파일 다운로드 가능하며, COCO Dataset 은 MS에서 공개해놓은 Traing Dataset 
  • weights 는 YOLO 모델 파일
  • cfg 는 Config 파일 

■ YOLOv3 네트워크 구조 

  • 입력값 : 416 x 416 [0,1] nomalization 되어 있으며 컬러 채널은 RGB 를 사용한다.  
  • 출력값 : 총 3군데 (82번째 출력, 94번째 출력, 106번째 출력)  
      - 조합해서 찾고자 하는 Object를 선정 한다. 
      - 507 X 85, 2028 X 85, 8112 X 85 형태의 Array 나옴 

■ YOLOv3 입력

  • SIZE : (320, 320), (416, 416), (608, 608)
  • Scale : 0.00392, (1/255)
  • Mean : [0, 0, 0]
  • RGB : true

■ YOLOv3 출력

  • 3개의 출력 레이어
    - outs[0].shape = (507,85), 507 = 13 * 13 * 3
    - outs[1].shape = (2028, 85), 2028 = 26 * 26 * 3
    - outs[2].shape = (8112, 85), 8112 = 52 * 52 * 3

■ YOLOv3 모델 & 설정 파일 다운로드

 

GitHub - pjreddie/darknet: Convolutional Neural Networks

Convolutional Neural Networks. Contribute to pjreddie/darknet development by creating an account on GitHub.

github.com

 

반응형

댓글