스테이블 디퓨전 설치하고 단순 이미지 출력에 만족하지 않아서
기존 누군가 만든 체크 포인트를 병합하는 것이 아닌
직접 이미지 자료들을 가져다 나만의 체크 포인트 파일을 만들기로 결심하였습니다.
이미지 학습 방법에는 로라, 하이퍼네트워크, 이런 것들이 있는데
그 중 가장 활용성 높고 광범위하게 이용 가능한 것이 드림부스라 하여 독학 하기 시작했습니다.
문제는 드림부스를 사용하는 사람이 너무나 없어서 자료가 충분치 않고
올라 온 자료도 사용 중 여러 문제를 해결해 주기에 너무나 부족하더군요.
몇 주 동안 외국 사이트부터 드림 부스 웹UI 만든 개발자랑 채팅도 하면서 여러가지 도움도 받으면서 얻게 된 정보들인데
너무나 기초적인 것들이라 말하기도 그렇지만 저 같은 초보 분들도 계실 것이라 생각 되어 몇 가지 적어 보자면,
스테이블 디퓨전 사용하다 이미지 출력에 문제가 있다거나 무언가 프로그램이 꼬인 것 같으면
스테이블 디퓨전 홈 폴더에서 config.json과 ui-config.json 파일을 지우고 프로그램을 모두 끈 다음 다시 실행하면 정상으로 되돌아 오더군요. 저는 이미지가 실사도 무조건 애니화 되는 현상을 경험하다 위 과정을 통해 해결했습니다.
또, 한 가지는 드림 부스를 스테이블 디퓨전에서 사용하다
역시 여러가지 설정이 꼬이거나 하여 문제가 생길 경우에
캐쉬 폴더 전체를 지우라고 하는데
위치는 C:\Users\(사용자가 설정한 이름)\.cache에 가시면 huggingface 라는 폴더가 있고 이를 삭제하시고 프로그램을 껏다가 다시 켜시면 됩니다.
마지막으로, 드림부스로 이미지 학습을 시키기 위해서
데이터 이미지들을 모았을때 대부분 사이즈가 512x512 인데
이 경우 사람의 경우 줌 아웃 화면에 어려움을 겪어 512x768 로 데이터 이미지를 만들기도 합니다.
이럴 경우 드림부스 화면에서 Model-Creat 항목에서 512x Model 을 선택하는 옵션이 있는데
아무리 데이터 이미지 사이즈가 768 이라고 하더라도 무조건 저 항목을 선택 하셔야 합니다.
선택을 하지 않을 경우 훈련 도중 저장 되는 ckpt 파일이 생성 되지를 않습니다.
아무래도 프로그램의 자잘한 버그 같은데 지금 이 문제가 해결 되지 않은 것 같습니다.
좋은 하루 되세요.
그대로 따라했는데 별 어려움이 없더군요.
학습 결과물을 보면 로라보다 훨 뛰어납니다. 실사 인물 학습에 사용했는데 여러 로라 화일들 올라와도서 비교해보면 직접 Checkpoint에 학습한 거 보다 여전히 못하더군요.
위에 2개 영상이 추천할 만 합니다.(전 처음에 왠 엔지니어가 올린 영상보고 하다가 고생했습니다.)
그리고 본인 사진을 찍어서 테스트 해보는 것이 가장 좋을 겁니다. 인풋이 좋아야 하는데 인터넷에 올라온 사진들은 편향적으로 잘나오는 각도가 주로 있죠. 칼라 밸런스 조정하고, 다양한 각도로 일정 한 사진을 찍어야 학습이 잘 됩니다. 전신 사진은 2장정도에 10장정도의 얼굴 위주 사진이면 됩니다.
또한 학습에 사용해도 되는 Checkpoint들이 있고 학습에 필요한 부분을 제거한 Checkpoint들도 있습니다.
기본이 되는 v1-5-pruned.ckpt 에 학습하는 것이 가장 좋을 겁니다.