kushagra124 commited on
Commit
e018390
·
1 Parent(s): 7c8f933

adding app with CLIP image segmentation

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -39,7 +39,6 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
39
 
40
  def visualize_images(image,predicted_images):
41
  alpha = 0.7
42
- # H,W = image.shape[:2]
43
  image_resize = cv2.resize(image,(352,352))
44
  resize_image_copy = image_resize.copy()
45
 
@@ -47,7 +46,6 @@ def visualize_images(image,predicted_images):
47
  resize_image_copy = cv2.addWeighted(resize_image_copy,alpha,mask_image,1-alpha,10)
48
 
49
  return cv2.convertScaleAbs(resize_image_copy, alpha=1.8, beta=15)
50
-
51
 
52
  def shot(image, labels_text):
53
  if "," in labels_text:
@@ -69,7 +67,7 @@ iface = gr.Interface(fn=shot,
69
  ["images/room.jpg","bed, table, plant, light, window,light"],
70
  ["images/image2.png","banner, building,door, sign,"],
71
  ["images/seats.jpg","door,table,chairs"],
72
- ["images/vegetables.jpg","carrot,radish,beans,potato,brnjal,basket"]
73
  ["images/room2.jpg","door,platns,dog,coffe table,mug,pillow,table lamp,carpet,pictures,door,clock"]
74
  ],
75
  # allow_flagging=False,
 
39
 
40
  def visualize_images(image,predicted_images):
41
  alpha = 0.7
 
42
  image_resize = cv2.resize(image,(352,352))
43
  resize_image_copy = image_resize.copy()
44
 
 
46
  resize_image_copy = cv2.addWeighted(resize_image_copy,alpha,mask_image,1-alpha,10)
47
 
48
  return cv2.convertScaleAbs(resize_image_copy, alpha=1.8, beta=15)
 
49
 
50
  def shot(image, labels_text):
51
  if "," in labels_text:
 
67
  ["images/room.jpg","bed, table, plant, light, window,light"],
68
  ["images/image2.png","banner, building,door, sign,"],
69
  ["images/seats.jpg","door,table,chairs"],
70
+ ["images/vegetables.jpg","carrot,radish,beans,potato,brnjal,basket"],
71
  ["images/room2.jpg","door,platns,dog,coffe table,mug,pillow,table lamp,carpet,pictures,door,clock"]
72
  ],
73
  # allow_flagging=False,