SSD 학습 방법

Question

안녕하세요. mmdetection에서 fast-train으로 학습을 시켜서 추론을 했습니다.

SSD도 동일하게 fast-train-kitty의 코드를 수정하여 학습이 가능한지 문의드립니다.

그리고 트레인, 밸리데이션, 테스트 이미지는 SSD300은 300*300, SSD512는 512*512로 맞춰주어야 하는지도 문의드립니다.(혹시 이미지 사이즈를 자동으로 변환해주는 소스코드도 mmdetection에서 제공되는지도 궁금합니다.)

Answer

안녕하세요, 인프런 AI 인턴이에요.

mmdetection에서 SSD를 학습하고자 할 경우, fast-train-kitty 코드를 수정하여 사용할 수 있습니다.

SSD300은 300*300, SSD512는 512*512 이미지를 입력으로 사용하기 때문에, 트레인, 밸리데이션, 테스트 이미지를 맞춰주어야 합니다.

또한, mmdetection에는 이미지 사이즈를 자동으로 변환해주는 소스코드는 제공되지 않습니다. 따라서, 해당 기능이 필요하다면 직접 소스코드를 작성해야 합니다.

위 내용이 도움이 되셨길 바라며, 추가적인 질문이 있다면 언제든지 댓글로 문의해주세요. 감사합니다.