[Object Detection] 4-5. YOLO v3 tiny

happy_quokka·2024년 1월 5일
0

딥러닝

목록 보기
16/18

가벼운 모델 만드는 방법

  • model depth를 줄인다 = layer를 줄인다
  • model width를 줄인다 = conv을 연산이 적은 conv을 사용한다 (예를 들어 5x5 conv이라면 3x3으로 줄인다)
  • input size를 줄인다
  • channel 수를 줄인다

yolov3 tiny

  • yolo v3 model 사이즈를 줄인 model
  • why?
    • 하드웨어에서 최대한 FPS 속도를 높이기 위해서이다
  • tiny는 max pooling을 많이 사용한다
  • model의 결과 yolo_result가 yolov3는 3개이고 tiny는 2개
  • activation의 경우 변경해가며 성능을 확인하면 좋지만 onnx나 tensorRT에서 지원하는지 확인해야한다

0개의 댓글