Yolo 모델 추론 결과 코드

안녕하세요.

Yolo 모델을 npu추론용으로 마련하는 과정은 다 완료하고 추론만 남았습니다.

Ultralytics 라이브러리에서 yolo 모델 추론을 지원하는 것처럼,

한 이미지에 대하여 object detection을 수행하고 추론된이미지(with box) 와 감지된 라벨 class 이름이 표현된 결과를 받는 코드를 찾고(짜고) 싶은데, 정확히 어디를 참고하면 될까요?

안녕하세요, 퓨리오사AI 최창민 입니다.

YOLO에 대해서 예시코드를 제공하고 있는 레포가 있어 안내드립니다.

warboy-vision-models 레포지토리를 참고해주시면 감사하겠습니다.

특히 언급하신 후처리 부분은 이 파일을 참고 부탁드립니다.

감사합니다.
최창민 드림

1 Like