CenterNet 질문입니다

Question

CenterNet 같은 경우도 같은 object 클래스에 대해서 중복되는 center 값을 예측할 수 있는 것 아닌가요?

AISchool · Answer

안녕하세요~. 반갑습니다. 말씀해주신대로 CenterNet의 경우에도 동일 Object에 대해 중복 Prediction이 발생할 수도 있습니다. (예를 들어, 1명의 사람을 검출할때 Center 포인트를 왼쪽다리, 오른쪽다리 이렇게 2개를 예측하면 1명의 사람에 대한 2개의 중복 Prediction이 발생할수도 있을 것입니다.) 다만 구조와 학습과정에서 기존의 2-stage Detector 계열은 학습과정에서 1개의 Object에 대해 여러개의 Bounding box를 Prediction하고 마지막에 Non Maximum Suppression으로 정제하는 것에 비해 (예를 들어, 1명의 사람에 대해서 100개의 Bounding box 예측값을 만들고 이중에서 Confidence가 높은 Bounding box Prediction을 남김) CenterNet은 학습과정에서 1명의 사람에 대해서 1개의 Center Point만을 예측하는 형태이기 때문에 중복 Prediction일 발생할 확률이 더 낮습니다. (즉, 학습과정에서 1개의 Object 에 대해 여러개의 Bounding box 예측후 정제 vs 1개의 Object에 대해 1개의 Center 포인트 예측의 차이입니다.) 좋은 하루되세요. 감사합니다.