import cv2 from segment_anything import SamPredictor mask_predictor = SamPredictor(sam) image_bgr = cv2.imread(IMAGE_PATH) image_rgb = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2RGB) mask_predictor.set_image(image_rgb) box = np.array([70, 247, 626, 926]) masks, scores, logits = mask_predictor.predict( box=box, multimask_output=True )