YOLO-World (오픈 보캐뷸러리 검출)

YOLO-World (오픈 보캐뷸러리 검출)

YOLO-World는 CVPR 2024에 채택된 실시간 오픈 보캐뷸러리 객체 검출기예요. Tencent AI Lab, ARC Lab(Tencent PCG), 화중과기대(HUST)가 함께 만들었어요. 고정된 클래스 대신 자연어 프롬프트(텍스트)로 검출할 클래스를 자유롭게 지정할 수 있어서 '열린 어휘(open-vocabulary)' 검출이 가능해요.

YOLO-World는 prompt-then-detect 패러다임을 써서, 어휘 임베딩을 모델 파라미터로 재파라미터화(reparameterize)해 추론 속도를 높여요. PyTorch 구현·사전훈련 가중치·Pretraining/Fine-tuning 코드가 전부 공개돼 있어요.

이 카테고리의 문서

  • 개요: 오픈 보캐뷸러리 검출과 prompt-then-detect
  • 논문: CVPR 2024와 YOLO-World 설계
  • 모델·사용: 사전훈련 가중치와 설치·파인튜닝

출처: https://github.com/AILab-CVC/YOLO-World