Skip to content

Commit efdbf02

Browse files
committed
fix docker
1 parent 5c74411 commit efdbf02

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

label_studio_ml/examples/segment_anything_2_image/Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ ENV PYTHONUNBUFFERED=1 \
2222
WORKERS=2 \
2323
THREADS=4 \
2424
CUDA_HOME=/usr/local/cuda \
25-
SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
2625

2726
# this image is already built with cuda 12.1.1, so we don't need to install it again
2827
# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
@@ -40,7 +39,7 @@ RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
4039
pip3 install -r requirements.txt
4140

4241
# install segment-anything-2
43-
RUN cd / && git clone --depth 1 --branch main --single-branch https://github.com/facebookresearch/segment-anything-2.git
42+
RUN cd ~ && git clone --depth 1 --branch main --single-branch https://github.com/facebookresearch/segment-anything-2.git
4443
WORKDIR /segment-anything-2
4544
RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
4645
pip3 install -e .

label_studio_ml/examples/segment_anything_2_image/README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,15 +136,17 @@ pip install -r requirements.txt
136136
2. Download [`segment-anything-2` repo](https://github.com/facebookresearch/sam2) into the root directory. Install SegmentAnything model and download checkpoints using [the official Meta documentation](https://github.com/facebookresearch/sam2?tab=readme-ov-file#installation)
137137
You should now have the following folder structure:
138138

139+
139140
| root directory
140141
| label-studio-ml-backend
141142
| label-studio-ml
142143
| examples
143-
| segment_anythng_2_image
144+
| segment_anything_2_image
144145
| sam2
145146
| sam2
146147
| checkpoints
147148

149+
148150
3. Then you can start the ML backend on the default port `9090`:
149151

150152
```bash

0 commit comments

Comments
 (0)