YOLO 대신 CenterNet을 선택하신 이유가 궁금합니다.

Question

안녕하세요. 강사님. 강의 잘 보고 있습니다. CenterNet에 대한 설명을 듣던 중 한 가지 의문점이 생겨 질문드립니다. YOLO 도 One-Stage model로 빠른 속도를 자랑하고 있는 데, One-Stage Model 에서 특별히 CenterNet을 채택하신 즈된 이유가 궁금합니다. Key-Point나 중복된 anchor prediction 방지를 위해서인지.. YOLO 모델이 이런 부분에선 부족한 것인지.. 의문이 들어서 질문 드립니다. ㅎㅎ 항상 감사드립니다!

AISchool · Answer

안녕하세요~. 반갑습니다. YOLO의 경우도 충분히 좋은 One-Stage Detector이며 사용해볼만한 선택지입니다. 다만 강의에서 주로 사용하는 TensorFlow Object Detection API 에서 YOLO 계열을 제공하지 않고 TensorFlow Object Detection API에서 지원하는 여러 모델중에서 경험상 CenterNet이 제일 괜찮은 결과를 보여주는 경우가 많아서 CenterNet을 사용하였습니다. 감사합니다~.