fix: update

fengx_launch.ipynb

@@ -33,7 +33,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [

fengxai/SelfSupervised.py

@@ -25,7 +25,7 @@ class SelfSupervised:
             "fengxai/config/GroundingDINO_SwinT_OGC.py", "weights/groundingdino_swint_ogc.pth")
 
     
-    def checkAnnotate(image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor, phrases: List[str]):
+    def checkAnnotate(self, image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor, phrases: List[str]):
         # 在原始图像中添加boxes
         h, w, _ = image_source.shape
         boxes = boxes * torch.Tensor([w, h, w, h])
@@ -49,7 +49,6 @@ class SelfSupervised:
     # 预测图片
     def imagePredict(self, imageFile, item="clothing", boxThreshold=0.3, textTreshold=0.25):
         from groundingdino.util.inference import load_image, predict
-        print("imgFile=", imageFile)
         src, img = load_image(imageFile)
         h, w, _ = src.shape
         boxes, logits, phrases = predict(

fengxai/inference.py

@@ -18,6 +18,5 @@ def imagePrediction(imageFile, funcCallback):
         os.system(cmd)
 
     selfSupervised = SelfSupervised()
-    print("imageFile=", imageFile)
     funcCallback(selfSupervised.imagePredict(imageFile))