diff --git a/.gitmodules b/.gitmodules index fddfa47..8e75d5d 100644 --- a/.gitmodules +++ b/.gitmodules @@ -10,3 +10,7 @@ path = pixsfm/Hierarchical-Localization url = https://github.com/cvg/Hierarchical-Localization.git branch = master +[submodule "muggled-sam/muggled_sam"] + path = muggled-sam/muggled_sam + url = https://github.com/heyoeyo/muggled_sam + branch = main diff --git a/muggled-sam/Dockerfile b/muggled-sam/Dockerfile new file mode 100644 index 0000000..32cd26d --- /dev/null +++ b/muggled-sam/Dockerfile @@ -0,0 +1,14 @@ +FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime + +RUN pip install \ + "opencv-python>=4.5,<4.12" \ + "numpy>=1.21,<2.3" + +RUN apt-get update && apt-get install -y \ + ffmpeg \ + libsm6 \ + libxext6 + +WORKDIR /workspace/muggled-sam + +CMD ["/bin/bash", "-l"] diff --git a/muggled-sam/docker-compose.yml b/muggled-sam/docker-compose.yml new file mode 100644 index 0000000..8c95ba7 --- /dev/null +++ b/muggled-sam/docker-compose.yml @@ -0,0 +1,16 @@ +services: + + muggled_sam: + build: + context: . + image: muggled_sam:dev + extends: + file: ../config.yml + service: win-gui-app + volumes: + - type: bind + source: ./muggled_sam + target: /workspace/muggled-sam + - type: bind + source: ../data/muggled-sam/saved_images + target: /workspace/muggled-sam/saved_images diff --git a/muggled-sam/muggled_sam b/muggled-sam/muggled_sam new file mode 160000 index 0000000..02bc10e --- /dev/null +++ b/muggled-sam/muggled_sam @@ -0,0 +1 @@ +Subproject commit 02bc10e6f9c550b8edc7b869f37d68ff809db23e diff --git a/muggled-sam/run-muggled-sam.bat b/muggled-sam/run-muggled-sam.bat new file mode 100644 index 0000000..f1cd39a --- /dev/null +++ b/muggled-sam/run-muggled-sam.bat @@ -0,0 +1 @@ +docker compose run --rm -it --name muggled_sam muggled_sam \ No newline at end of file