[스테이블 디퓨전] Rembg 확장 기능으로 이미지 배경 누끼 따기

Rembg 확장 기능이란?

스테이블 디퓨전에는 배경 누끼를 따는 확장 기능이 몇 개가 있다.

그중에 먼저 Rembg 기능을 알아 보자!

물론, 배경 누끼를 따주는 전문화된 사이트도 많다.

오히려 그런 곳이 더 누끼를 잘 따주기도 한다.

그래도 스테이블 디퓨전에 이런 기능도 있다!

정도만 알고 가끔은 사용하는데 오히려 편한 것도 있다.

Rembg 설치하기

설치는 쉽다. 늘 하듯 URL로 설치하면 된다.

[Extensions] – [Install from URL] – URL for extension’s git repository에 아래 URL 입력

https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg.git

GitHub – AUTOMATIC1111/stable-diffusion-webui-rembg: Removes backgrounds from pictures. Extension for webui.

Removes backgrounds from pictures. Extension for webui. – GitHub – AUTOMATIC1111/stable-diffusion-webui-rembg: Removes backgrounds from pictures. Extension for webui.

설치 완료 후 재시작하면 [Extras] 탭 하단에 Remove background가 만들어진 것을 볼 수 있다.

이 Rembg 확장 기능은 [Extras] 탭에서만 사용 가능하다.

Rembg로 배경 누끼 따보자

사용법은 간단하다.

 [Extras] 탭에 이미지를 넣고 하단 Remove background 메뉴에서 원하는 방식으로 선택 후

생성을 누르면 누끼가 따진다.

아래가 원본이미지

아래가 누끼를 딴 모습

누끼 따는 방식 종류의 차이

메뉴를 보면 누끼 따는 방식이 5가지가 제공된다.

각각의 차이가 조금 있다.

u2 net : A pre-trained model for general use cases.
u2 netp : A lightweight version of u2net model.
u2 net_human_seg : A pre-trained model for human segmentation.
u2 net_cloth_seg: A pre-trained model for Cloths Parsing from human portrait. 
silueta: Same as u2net but the size is reduced to 43Mb.

즉, u2 net이 가장 일반적인 학습 모델인데, u2 netp이 바로 조금 경량화 한 버전이라는 점

그래서 딴 거는 말고 일반적으로 u2 netp을 사용하면 되고 

조금 복잡한 인물의 경우는 u2 net_human_seg을 사용하면 됨