Run Segment Anything Model 2 on a live video stream
# SAM2 live video stream # SAM2 streaming # real-time SAM2 # webcam SAM2
- 27/11/2024 : 최초 SAM2 실시간 세그멘테이션 코드 성공
- sam2, sam2.1 둘다 가능
반드시 파이썬 버전은 3.11 이상이어야 합니다
conda create -n seg python=3.11 -y
conda activate seg
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -e .
pip install -e ".[notebooks]"
위에 명령어 후 아래와 같은 오류 발생시
File "/tmp/pip-build-env-v31jxhmj/overlay/lib/python3.11/site-packages/torch/__init__.py", line 367, in <module>
from torch._C import * # noqa: F403
^^^^^^^^^^^^^^^^^^^^^^
ImportError: /tmp/pip-build-env-v31jxhmj/overlay/lib/python3.11/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkComplete_12_4, version libnvJitLink.so.12
[end of output]
note: This error originates from a subprocess, and is likely not a problem with pip.
error: subprocess-exited-with-error
× Getting requirements to build editable did not run successfully.
│ exit code: 1
╰─> See above for output.
note: This error originates from a subprocess, and is likely not a problem with pip.
다음과 같은 명령어 진행
python setup.py build_ext --inplace다음으로 필요 라이브러리 설치
pip install -r requirements.txt
SAM2 모델 다운
cd checkpoints/sam2
./download_ckpts.sh
cd ../..SAM2.1 모델 다운
cd checkpoints/sam2.1
./download_ckpts.sh
cd ../..자 이제 SAM2를 실시간 스트리밍 화면에 적용할 준비가 됐습니다.
- mp4 파일에서 테스트 해보기
python demo.py- webcam에서 실시간 sam2 적용하기 - 마우스로 바운딩 박스를 그려서 sam2할 객체 지정
python demo_webcam_box.py
- webcam에서 실시간 sam2 적용하기 - 마우스 point 클릭으로 sam2할 객체 지정
python demo_webcam_point.py
- webcam에서 실시간 sam2 적용하기 - 첫 프레임에 사람 tracking
python demo_webcam_yolo.py