Abijith commited on
Commit
91b984b
1 Parent(s): a3b4745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ def segment_to_bbox(segment_indexs):
47
  return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
48
 
49
  def clipseg_prediction(image):
50
-
 
51
  img_w, img_h = image.shape
52
  inputs = clip_processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
53
  # predict
 
47
  return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
48
 
49
  def clipseg_prediction(image):
50
+
51
+ print('image_shape:: ', image.shape)
52
  img_w, img_h = image.shape
53
  inputs = clip_processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
54
  # predict