티스토리 뷰

IT 정보공간

스테이블 디퓨전 PNG 이미지에서 프롬프트 추출하는 방법

여행인생 2023. 5. 18. 23:33

스테이블 디퓨전으로 이미지를 만들어내면서 어떤 프롬프트를 사용할지 고민이 많으시죠? 그래서 PNG 이미지 파일에서 프롬프트를 추출하는 방법을 공유해 드립니다. 추출한 포인트로 앞으로 더 쉽게 원하시는 이미지를 뽑을 수 있게 도움이 되리라 믿습니다.

PNG 이미지에서 프롬프트 추출하는 방법

몇 가지 방법이 있는데요. 오늘은 2가지 방법을 쉽게 알려드리겠습니다. 혹시 컴퓨터 사양이 좋지 않아서 스테이블 디퓨전을 하기 어려우신 분들은 필자의 이지 스테이블 디퓨전 코랩을 이용해서 시작할 수 있습니다. 코랩 설치 방법은 링크를 참고 부탁드립니다.

1 스테이블 디퓨전 확장기능 이용

코랩으로 이지 스테이블 디퓨전을 실행합니다. 실행하고 나서 상단 메뉴 가장 우측 끝에 있는 Extensions를 클릭합니다. 익스텐션 하단에 Install from URL이라는 메뉴를 클릭해 줍니다. 그럼 아래 이미지와 같이 3줄이 나오는데요. URL for extension's git repository에 아래 주소를 복사해서 붙여 넣습니다. 그리고 가장 아래 Install 버튼을 클릭해 주면 끝입니다. 인스털 버튼 아래에 설치된 경로가 나오죠?

https://huggingface.co/spaces/hysts/DeepDanbooru

DeepDanbooru - a Hugging Face Space by hysts

huggingface.co

그래도 궁금하시다면 Installed 메뉴를 클릭해서 DeepDanbooru가 설치되었는지 확인하시면 됩니다.

이제 image to image 메뉴로 옮겨가셔서 사진을 먼저 업로드합니다. 그리고 Interrogate DeepBoru를 클릭하시며 프롬프트를 확인하실 수 있습니다. 쉽죠?

1 hugging face 이용

프롬프트를 찾는 용도로 많이 찾는 웹사이트인데요. 아래 링크를 이용하시면 굳이 스테이블 디퓨전에 확장프로그램을 설치하지 않아도 프롬프트를 확인하실 수 있습니다.

https://huggingface.co/spaces/hysts/DeepDanbooru

사용하는 방법은 비슷한데요. 링크에 들어가셔서 프롬프트를 추출하실 이미지를 끌어서 업로드합니다. 그리고 Score Threshold를 조절해서 Submit를 클릭하면 아래 이미지와 같이 Output에서 프롬프트를 확인하실 수 있습니다. 이미지는 필자가 파리 여행을 시나리오로 만든 AI 룩북 여행 사진인데요. 실제 같죠? 스테이블 디퓨전은 프롬프트를 자세히 넣을수록 원하는 이미지를 얻으실 확률이 높아집니다. 많이 해보시면 재미도 있고요.