Stable Diffusion GUI 중 하나인 ComfyUI에서
이미지 배경제거 모듈인 Rembg를 사용하기 위해 커스텀 노드를 개발해보았다.
개발하게 된 연유는 다음과 같다.
스테이블 디퓨전을 사용해서 Gif 애니메이션을 제작하려는데...
배경을 제거하지 않고 Openpose, Depth, LineArt 등의 Controlnet Preprocessing을
하게 되면 불필요한 요소를 인식하는 탓에, 깔끔하게 이미지 전처리가 되지 않았던것...
기존에 이미 해당 기능을 제공하는 Git repo가 있었지만, 조금 아쉬웠던 점은
단일한 이미지가 아닌, 이미지 배치를 input으로 주게 되면, 
다음과 같은 에러가 발생하게 된다.. 8개의 input 이미지를 주었으나,
코드에는 1개의 단일 이미지만 처리하게끔 되어 있다.
그러나 애니메이션을 제작하기 위해서는 여러 장의 이미지도 처리할 수 있어야
하는 법.. 그런고로 복수 이미지 처리가 가능하게끔 기존 코드를 개선시켜보았다.
결과 :
영상에서 간단하게 gif를 뽑는 용도로도 쓸 수 있고....
컨트롤넷 과 결합하여 동영상에서
"이미지 추출 - 이미지 전처리(depth, openpose 등) - 컨트롤넷 적용 - gif 이미지 생성"
과 같은 workflow를 가질 수도 있다.
끗!
링크 :
https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes
한 줄 한 줄, 매일 매일