레오나르도AI 파인튜닝 학습모델 만들어 보기

 

오늘의 글 한줄 요약

학습 데이터의 노멀라이즈를 잘 해줍시다.

 

단계별 요약

1. 이미지를 준비합니다.

2. 512x512 pixel 로 해상도 (width, height)를 맞춥니다.

3. 학습할 물체가 중앙에 오게 위치 (x,y)를 맞춥니다.

4. 학습할 물체의 크기가 비슷하도록 크기 object size(width, height)을 맞춥니다.

5. 물체의 색감을 맞춥니다. hue/saturation

6. 조명이 너무 엇나간 부분을 맞춥니다.

 

내용이 길어서 앞에 썰이 길었습니다. 거두절미하고 빠르게

시작합니다.

1. 데이터를 폴더에 몰아넣고 준비합니다. 해상도가 들쑥 날쑥 합니다. 이러면 Stable Diffusion 이 싫어해요.

 

2. 포토샵을 이용하였스빈다. 학습할 물체를 일관되게 중앙에 오게 하기 위해 가이드 라인을 잡아줍니다.

 

3. Stable Diffusion v1.5 가 좋아하는 512 x 512 pixel 로 리사이징 해줍니다. 이미지 사이즈(Image Size)로 조절하면 스테레칭 되 듯 늘어나거나 줄어드니 캔버스 사이즈(Canvas Size)로 조절합니다.

 

4. 가로:세로 비율을 1:1로 캔버스를 조절하고 나면 빈공간이 생깁니다. 이 공간을 매직툴(Magic Tool)로 선택하고 살짝 확장 선택(Select > Modify > Expand)해 줍니다. 여기에서 이 이미지에는 2 pixel 정도 적당하겠다, 이 이미지는 10 pixel 정도는 해줘야 겠는 걸 하는 경험적 값이 필요합니다. 이번 이미지는 단순하기에 2 pixel 로 줬습니다.

 

 

5. 포토샵의 채워주기 마법 들어갑니다. Edit > Content-Aware Fill 로 채워줍니다.

 

6. 중간 점검 짠!

 

7. 이제 가로:세로 비율이 1:1이 되었으니 Image Size로 512 x 512 pixel 로 리사이징합니다.

 

8. 색감이 어두는 이미지들도 있습니다. Image > Adjustment > Levels로 살펴보니 최대 밝기 픽셀이 255까지 도달하지 못했습니다.

 

9. Image > Adjustment > Levels 로 조절해줍니다. 물론 255에 딱 닿으면 그것 또한 자연스럽지 못하기에 살짝, 아주 살짝 닿게 해줍니다. 소프트 터치!

 

10. 이번에는 볼륨감을 맞춰줄 차례입니다. 게임에 들어가는 스프라이트 이미지 노동 작업 하는 줄 알았습니다. 이미지 내에서 학습할 물체가 차지하는 object size의 scale를 맞추는 작업입니다.

 

11. 포토샵 레이어를 기준이 될 물체 레이어는 100%, 맞출 물체 레이어는 50%로 해서 반투명하게 놓고서 맞춥니다.

 

12. 창문의 크기, 문의 크기, 계단의 크기, 소파의 크기 등을 보이는 단서들로 사이즈를 가늠해가며 스케일을 맞춥니다.

 

13. 짠, 512 x 512 pixel 로 노멀라이즈를 마쳤습니다.

 

14. 이제 마감하려는데 한 이미지가 눈에 밟힙니다. 뭔가 데이터셋에 outlier가 껴있는 듯합니다. 손님 샵 마감시간 다되갑니다.

 

15. 잠시만요, 이것만 좀 더 하고 금방 갈게요. 창문을 잡습니다.

 

16. Hue/Saturation 을 잡아줍니다. 살짝 노랗게, 살짝 밝게.

 

17. 저장을 하고 유유히 샵을 빠져나옵니다. 노멀라이즈 된 데이터셋 짠! 속이 개운합니다. 내 다음엔 다른 샵에 가야지 하고서 20년 넘게 다니는 단골 샵입니다.

 

18. 레오나르도 AI 로 들어와서 노멀라이즈 된 데이터셋을 올립니다. 이 때 부터 두근거리기 시작합니다. 이번엔 잘 나올까.

 

19. 나머지 절차들을 수행합니다. Base Model 은 Stable Diffusion v1.5 (512x512) 모델, 이것을 겨냥하고 여기까지 왔습니다.

 

20. 카테고리 타입이 어떤 것들이 있나 구경도 해봅니다. 이거 지정하면 학습이 더 잘될까 하는 기대감도 가져보지만 아직은 아닌걸로 추정합니다.

 

21. 학습을 기다립니다.

 

22. 그래서 오늘의 결과는 짠! 만족스럽습니다.