Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
@cxnt
cxnt authored Oct 8, 2024
1 parent 46b02c6 commit 2b64680
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions docker/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,13 @@ RUN pip3 install opencv-python

RUN pip3 install mock==5.1.0

RUN mkdir sam2.1_weights
RUN mkdir /sam2.1_weights
# Create directory for SAM weights
RUN mkdir -p /sam2.1_weights

# Download SAM 2.1 checkpoints
RUN curl -o /sam2_weights/sam2.1_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
RUN curl -o /sam2_weights/sam2.1_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt"
RUN curl -o /sam2_weights/sam2.1_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt"
RUN curl -o /sam2_weights/sam2.1_hiera_large.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt"
RUN curl -L -o /sam2.1_weights/sam2.1_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
RUN curl -L -o /sam2.1_weights/sam2.1_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt"
RUN curl -L -o /sam2.1_weights/sam2.1_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt"
RUN curl -L -o /sam2.1_weights/sam2.1_hiera_large.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt"

LABEL python_sdk_version=6.73.203

0 comments on commit 2b64680

Please sign in to comment.