Add segment_anything package. (#1241)

Code for running inference with the SegmentAnything model: https://github.com/facebookresearch/segment-anything

http://b/277217745

Author: rosbo

Dockerfile.tmpl

@@ -547,6 +547,7 @@ RUN pip install pytorch-ignite \
         openpyxl \
         timm \
         torchinfo && \
+        pip install git+https://github.com/facebookresearch/segment-anything.git && \
     /tmp/clean-layer.sh
 
 # Install pycolmap for py3.10 (only the conda version works).

tests/test_imports.py

@@ -7,3 +7,4 @@ def test_basic(self):
         import cleverhans
         import tensorflow_gcs_config
         import tensorflow_datasets
+        import segment_anything