관리 메뉴

가끔 보자, 하늘.

Stable Zero123 설치하기 본문

개발 이야기/개발툴 및 기타 이야기

Stable Zero123 설치하기

가온아 2023. 12. 14. 12:05

https://stability.ai/news/stable-zero123-3d-generation?fbclid=IwAR3XuAFOQzLizSvOyMnvfb7NT0XHmaVRg_f7n8bxSUzCZ7gYexBjtA9-ySA

 

Introducing Stable Zero123: Quality 3D Object Generation from Single Images — Stability AI

Stable Zero123 is an AI-powered model for generating novel views of 3D objects with improved quality. Released for non-commercial and research purposes, it uses an improved dataset and elevation conditioning for higher-quality predictions. Using the improv

stability.ai

이미지 한 장으로 3D 모델을 만들어주는 모델이 릴리즈 되었습니다. 기존에도 여럿 비슷한 모델이 있었으나 이번에 나온 모델은 뒷면까지도 깔끔하게 뽑아 준다고 하네요. sd1.5로 이미지를 생성 후 이를 3d 모델로 전환한다고 합니다.

각설하고 바로 설치해보죠. 우선 이 링크에서 모델을 다운받아 두세요. 그리고 threestudio github 페이지로 이동하여 visual studio로 열어 줍니다.

three studio는 webui 와 유사한 역할을 합니다. 2d 이미지 대신 3d 생성 모델을 설정해 3d 모델링된 데이터를 생성해주는 UI역할을 합니다. 현재는 프리뷰 버전이라 webui 와 같이 편리한 배포 환경을 갖춘 상태는 아닙니다. 그래서 소스 코드를 직접 빌드해서 실행해야 합니다.

설치하는 과정에 여러 에러가 발생할 수 있습니다. ModuleNotFoundError: No module named 'torch' 이라던가 cannot import name 'Schema' from 'pydantic' 같은 에러가 발생할 수 있는데 현재 몇몇 패키지의 의존성이 꼬여 있습니다.

문제가 되는 패키지들을 아래 버전으로 맞춰 설치하시면 에러가 해결될 수 있으며 진행이 안될 경우 에러 메세지 전체와 현재 버전들을 정리해서 댓글을 달아주시면 확인해 보겠습니다.

  • fastapi 0.88.0
  • pydantic 1.10.6
  • gradio 3.36.1
  • opencv-python 4.7.0.68
  • torch와 torchvision은 현재 자신의 cu 버전에 맞게 설치되어 있으면 됩니다.

가상환경에서 의존성 맞추려고 손대봤는데... 그럴수도 더 꼬이네요. requirements.txt 에서 위 내용으로 버전 맞추로 한방에 설치 시도하시는게 좋습니다.

만약 정상적으로 실행되었다면 아래와 같은 화면을 보실 수 있습니다만... 

전.... 4080 T-T ... 16GB에서는 아직 실행 불가합니다. 4090 유저분들은 좋으시겠어요... 어흑!!

끝!!

반응형