본문 바로가기
딥러닝 어쩌구/연구일지&디버깅

[디버깅] Streamlit 데모 만들기...

by 포숑은 맛있어 2023. 11. 28.
반응형

목표: Streamlit으로 SAM, VOS, VI 모두 돌아가게 해보자

1. 클릭 관련
- pip install streamlit-image-coordinates
- cv2로 클릭받는건 UI backend 있어야 가능. GPU 서버에서는 X

 

2. 모델 실행

모델들 돌아갈 수 있게 도커 이미지 하나로 통합하고, streamlit 버튼으로 subprocess.command()로 실행시킴


3. ffmpeg libx264 에러

- 결과 영상 cv2로 만드니까 코덱문제 생겨서 web streamable한 libx264로 바꿔줘야함. 근데 unknown error 뜸

- 검색해서 뜨는거 다 안되길래 챗지피티의 힘을 빌렸으나.. 안됨. 그냥 다 다시 깔아야겠음

- 이거 처음부터 전부 따라하면 될듯

- https://gist.github.com/iJackUA/5582974    

- https://askubuntu.com/questions/148554/unknown-option-enable-libfaad-error-libfaac-not-found-while-configuring-f

둘다 그게 그거니까 암거나...

 

 

git clone --depth 1 git://source.ffmpeg.org/ffmpeg

근데 이게 안됨.

git clone https://git.ffmpeg.org/ffmpeg.git ffmpeg

이걸로 대신함

 

 --enable-libfaac --enable-libfdk-aac 이 옵션 2개도 안됨.

중요한건 --enable-libx264이랑 gpl이니까 걍 지우고 시도해보자.

 

지금 make만 몇번째인지...

 

add-apt-repository

apt-get install software-properties-common

 

...

 

깔끔하게 지웠다 깔아도 libx264 안됨

make install 안했다는 사람

https://unix.stackexchange.com/questions/149091/how-to-add-aac-and-libx264-to-ffmpeg-installation      

 

 

--------------------------

걍 도커 이미지 새로 만들었음. 만드는거 참고

https://jjeongil.tistory.com/1588                           

 

endpoint.. cmd... docker 명령어를 알아야겠다

# app/Dockerfile

FROM python:3.9-slim

WORKDIR /app

RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    software-properties-common \
    git \
    && rm -rf /var/lib/apt/lists/*

RUN apt-get -y update
RUN apt-get -y upgrade
RUN apt-get install -y ffmpeg

RUN git clone https://github.com/streamlit/streamlit-example.git .

RUN pip install opencv-contrib-python
RUN pip install opencv-python
RUN pip3 install -r requirements.txt
RUN pip3 install matplotlib
RUN pip3 install networkx
RUN pip install scikit-learn  --user --upgrade
RUN pip install scikit-image==0.16.2
RUN pip install mmcv-full

EXPOSE 8505
CMD ["bash"]

 

명령어 실행

cd /workspace
cd segment-anything; pip install -e .
pip install opencv-python pycocotools matplotlib onnxruntime onnx

cd ../../Xmem; pip install -r requirements.txt 
pip install torch==2.0.0 torchaudio==2.0.0 torchdata==0.6.0 torchmetrics==1.1.2 torchtext==0.15.1 torchvision==0.15.1

streamlit run cg_demo_stream.py --server.port=8505

 

반응형

댓글