SysCV · rtyley · Oct 22, 2023 · rtyley · Oct 23, 2023 · rtyley
diff --git a/demo/demo_hqsam.py b/demo/demo_hqsam.py
@@ -2,7 +2,7 @@
 import torch
 import matplotlib.pyplot as plt
 import cv2
-from segment_anything import sam_model_registry, SamPredictor
+from segment_anything_hq import sam_model_registry, SamPredictor
 from segment_anything_hq import sam_model_registry 
 from segment_anything_hq import sam_model_registry 
 import os
 
 def show_mask(mask, ax, random_color=False):

diff --git a/demo/demo_hqsam_light.py b/demo/demo_hqsam_light.py
@@ -2,7 +2,7 @@
 import torch
 import matplotlib.pyplot as plt
 import cv2
-from segment_anything import sam_model_registry, SamPredictor
+from segment_anything_hq import sam_model_registry, SamPredictor
 import os
 
 def show_mask(mask, ax, random_color=False):

diff --git a/segment_anything/__init__.py → segment_anything_hq/__init__.py b/segment_anything/__init__.py → segment_anything_hq/__init__.py
diff --git a/segment_anything/automatic_mask_generator.py → ...t_anything_hq/automatic_mask_generator.py b/segment_anything/automatic_mask_generator.py → ...t_anything_hq/automatic_mask_generator.py
diff --git a/segment_anything/build_sam.py → segment_anything_hq/build_sam.py b/segment_anything/build_sam.py → segment_anything_hq/build_sam.py
diff --git a/segment_anything/build_sam_baseline.py → segment_anything_hq/build_sam_baseline.py b/segment_anything/build_sam_baseline.py → segment_anything_hq/build_sam_baseline.py
diff --git a/segment_anything/modeling/__init__.py → segment_anything_hq/modeling/__init__.py b/segment_anything/modeling/__init__.py → segment_anything_hq/modeling/__init__.py
diff --git a/segment_anything/modeling/common.py → segment_anything_hq/modeling/common.py b/segment_anything/modeling/common.py → segment_anything_hq/modeling/common.py
diff --git a/segment_anything/modeling/image_encoder.py → ...ent_anything_hq/modeling/image_encoder.py b/segment_anything/modeling/image_encoder.py → ...ent_anything_hq/modeling/image_encoder.py
diff --git a/segment_anything/modeling/mask_decoder.py → segment_anything_hq/modeling/mask_decoder.py b/segment_anything/modeling/mask_decoder.py → segment_anything_hq/modeling/mask_decoder.py
diff --git a/segment_anything/modeling/mask_decoder_hq.py → ...t_anything_hq/modeling/mask_decoder_hq.py b/segment_anything/modeling/mask_decoder_hq.py → ...t_anything_hq/modeling/mask_decoder_hq.py
diff --git a/segment_anything/modeling/prompt_encoder.py → ...nt_anything_hq/modeling/prompt_encoder.py b/segment_anything/modeling/prompt_encoder.py → ...nt_anything_hq/modeling/prompt_encoder.py
diff --git a/segment_anything/modeling/sam.py → segment_anything_hq/modeling/sam.py b/segment_anything/modeling/sam.py → segment_anything_hq/modeling/sam.py
diff --git a/segment_anything/modeling/tiny_vit_sam.py → segment_anything_hq/modeling/tiny_vit_sam.py b/segment_anything/modeling/tiny_vit_sam.py → segment_anything_hq/modeling/tiny_vit_sam.py
diff --git a/segment_anything/modeling/transformer.py → segment_anything_hq/modeling/transformer.py b/segment_anything/modeling/transformer.py → segment_anything_hq/modeling/transformer.py
diff --git a/segment_anything/predictor.py → segment_anything_hq/predictor.py b/segment_anything/predictor.py → segment_anything_hq/predictor.py
diff --git a/segment_anything/utils/__init__.py → segment_anything_hq/utils/__init__.py b/segment_anything/utils/__init__.py → segment_anything_hq/utils/__init__.py
diff --git a/segment_anything/utils/amg.py → segment_anything_hq/utils/amg.py b/segment_anything/utils/amg.py → segment_anything_hq/utils/amg.py
diff --git a/segment_anything/utils/onnx.py → segment_anything_hq/utils/onnx.py b/segment_anything/utils/onnx.py → segment_anything_hq/utils/onnx.py
diff --git a/segment_anything/utils/transforms.py → segment_anything_hq/utils/transforms.py b/segment_anything/utils/transforms.py → segment_anything_hq/utils/transforms.py
diff --git a/setup.cfg b/setup.cfg
@@ -4,7 +4,7 @@ multi_line_output=3
 include_trailing_comma=True
 known_standard_library=numpy,setuptools
 skip_glob=*/__init__.py
-known_myself=segment_anything
+known_myself=segment_anything_hq
 known_third_party=matplotlib,cv2,torch,torchvision,pycocotools,onnx,black,isort
 no_lines_before=STDLIB,THIRDPARTY
 sections=FUTURE,STDLIB,THIRDPARTY,MYSELF,FIRSTPARTY,LOCALFOLDER

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 from setuptools import find_packages, setup
 
 setup(
-    name="segment_anything",
+    name="segment_anything_hq",
     version="1.0",
     install_requires=[],
     packages=find_packages(exclude="notebooks"),