fix: update

fengxai/SelfSupervised.py

@@ -4,13 +4,13 @@ import cv2
 import numpy as np
 from segment_anything import SamPredictor, sam_model_registry
 from diffusers import StableDiffusionInpaintPipeline
-from groundingdino.util.inference import load_model, load_image, predict, annotate
 from torchvision.ops import box_convert
 from typing import List
 
 class SelfSupervised:
-    
     def __init__(self):
+        from groundingdino.util.inference import load_model
+        
         # -----Set Image and CUDA
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         # ------SAM Parameters
@@ -48,6 +48,8 @@ class SelfSupervised:
     
     # 预测图片
     def imagePredict(self, imgFile, item="clothing", boxThreshold=0.3, textTreshold=0.25):
+        from groundingdino.util.inference import load_image, predict
+        
         src, img = load_image(imgFile)
         h, w, _ = src.shape
         boxes, logits, phrases = predict(

fengxai/inference.py

@@ -6,7 +6,7 @@ def imagePrediction(imageFile, funcCallback):
         and not os.path.exists("weights/groundingdino_swint_ogc.pth"):
         cmd = "pip install git+https://github.com/IDEA-Research/GroundingDINO.git"
         os.system(cmd)
-            
+        
         cmd = """
             mkdir weights \
                 && cd weights \