Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -284,11 +284,11 @@ class SAMPointVisualizationDemo(object):
         height, width, _ = ori_image.shape
 
         image = ori_image
-        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        ori_image = cv2.cvtColor(ori_image, cv2.COLOR_BGR2RGB)
+        # image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        # ori_image = cv2.cvtColor(ori_image, cv2.COLOR_BGR2RGB)
 
         input_point = np.array(points)
-        input_label = np.array([1])
+        input_label = np.array([1] * len(points))
 
         with torch.no_grad():
             self.predictor.set_image(image)