컬러와 그레이스케일 이미지가 섞였을 때 모델 Input을 어떻게 설정하면 좋을까요?

Question

안녕하세요. 개인 데이터셋을 훈련을 시키려고 하는데, 이게 그레이스케일 이미지입니다

개인 데이터셋이 충분치 않아 비슷한 공개된 이미지들을 훈련시켜서 이를 pretrained 모델로 사용하고자 합니다. 다만 공개된 이미지는 3채널 이미지라서 고민이 됩니다

제가 생각하기에 2가지 방향이 있는데, 보통 방법은 어떻게 하는지 알고 싶습니다

1. 공개 데이터를 grayscale로 바꿔서 모델 Input을 채널 1로 하는것

2. 공개 데이터를 기존 rgb로 훈련한 후 개인 데이터셋에 대해서 3채널로 바꿔서 훈련하는 것

권 철민 · Answer

안녕하십니까,

음, 공개된 이미지가 데이터 량이 많고 inrference에 들어오는 데이터가 공개된 이미지와 비슷하다면 3채널 데이터로 학습 하고, 그렇지 않다면 gray scale로 학습 데이터를 모두 바꿔서 학습하고, 테스트 해보는게 좋을 것 같습니다.

감사합니다.

유영재 · Answer

감사합니다

인프런 커뮤니티 질문&답변