aiprofile-gdsc-koreauniv / ai-api-server

AI Generation API Server for ai-profile project
0 stars 2 forks source link

Feature/face preprocess/feat: Add handling Non-RGB channel #4

Closed seooyxx closed 7 months ago

seooyxx commented 7 months ago

Motivation

전처리 모듈 중 입력 사진이 JPG가 아닐 경우, 특히 PNG 사진 등에서 RGBA 채널이 전달되는 경우 face_detector (YOLO) 가 터지는 오류를 테스트 중 발견했습니다. AI 서버 이전 단계에서 JPG 변환기가 작동하지만, 해당 변환기의 오류 가능성 및 AI 로직의 강건함을 위해 RGB 및 GRAYSCALE 이미지를 모두 RGB 채널로 변경하도록 코드를 추가했습니다.

Key Changes

convert_to_rgb 함수 추가

Usage Example

preprocess_image 전체 로직 중에 다음 코드 부분만 변경했습니다.

Before:

ndarr_images = [np.array(image) for image in images]    

After:

ndarr_images = [convert_to_rgb(np.array(image)) for image in images]    

올바르게 작동한다면, 추가적인 수정은 필요하지 않습니다.

To Reviewers

opencv만 필요하기 때문에 기존 종속성 수정 필요 없습니다. 올바르게 작동하는지만 점검해 주시면 될 것 같습니다. 우선 AI 서버 이전 JPG 변환기가 올바르게 작동해야 합니다.