Update squeeze_sam until further refactoring.

clxie · Dec 18, 2023 · 8f9ba7a · 8f9ba7a
1 parent 3a0cc80
commit 8f9ba7a
Show file tree

Hide file tree

Showing 10 changed files with 2 additions and 1,031 deletions.
diff --git a/EfficientSAM_example.py b/EfficientSAM_example.py
@@ -1,5 +1,5 @@
 from efficient_sam.build_efficient_sam import build_efficient_sam_vitt, build_efficient_sam_vits
-from squeeze_sam.build_squeeze_sam import build_squeeze_sam
+# from squeeze_sam.build_squeeze_sam import build_squeeze_sam
 
 from PIL import Image
 from torchvision import transforms
@@ -21,7 +21,7 @@
 models['efficientsam_s'] = build_efficient_sam_vits()
 
 # Build the SqueezeSAM model.
-models['squeeze_sam'] = build_squeeze_sam()
+# models['squeeze_sam'] = build_squeeze_sam()
 
 # load an image
 sample_image_np = np.array(Image.open("figs/examples/dogs.jpg"))

diff --git a/squeeze_sam/__init__.py b/squeeze_sam/__init__.py
diff --git a/squeeze_sam/build_squeeze_sam.py b/squeeze_sam/build_squeeze_sam.py
diff --git a/squeeze_sam/detection_transformer.py b/squeeze_sam/detection_transformer.py