yolov10 모델 변환 및 NPU 포팅(1)

HeungJun Kim·2024년 5월 31일
0

YOLO

목록 보기
7/7

yolov10 repo

https://github.com/THU-MIG/yolov10

yolv10 특징

  • unsupported operators:['TopK', 'GatherElements', 'Mod']

yolov10에는 NMS가 없기때문에, TOPK 300개의 object를 추출하는 레이어가 포함되어있다.

  • onnx 모델에서 미지원 operator 노드를 제거하고 후처리를 직접 구현해줘야한다.
    https://github.com/THU-MIG/yolov10/issues/75
  • 더 쉬운 방법은 yolov10s 모델 기준 '/model.23/Transpose_output_0' 을 파싱하면된다.
    (Anchor Free 기반 yolo 모델의 output과 동일함, gird by 1 anchor)

profile
Computer Vision / ADAS / DMS / Face Recognition

4개의 댓글

comment-user-thumbnail
2024년 7월 15일

혹시, DMS/OMS 관련 개발을 지금도 하고 계신가요?

1개의 답글