kushagra124 commited on
Commit
183692d
1 Parent(s): bba5a32

adding app with CLIP image segmentation

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -37,7 +37,7 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
37
 
38
  return predicted_masks
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()
@@ -45,9 +45,9 @@ def visualize_images(image,predicted_images):
45
  for mask_image in predicted_images:
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:
52
  prompts = labels_text.split(',')
53
  else:
@@ -55,7 +55,7 @@ def shot(image, labels_text):
55
  prompts = list(map(lambda x: x.strip(),prompts))
56
  predicted_images = detect_using_clip(image,prompts=prompts)
57
 
58
- category_image = visualize_images(image=image,predicted_images=predicted_images)
59
  return category_image
60
 
61
  iface = gr.Interface(fn=shot,
 
37
 
38
  return predicted_masks
39
 
40
+ def visualize_images(image,predicted_images,brightness=15):
41
  alpha = 0.7
42
  image_resize = cv2.resize(image,(352,352))
43
  resize_image_copy = image_resize.copy()
 
45
  for mask_image in predicted_images:
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=brightness)
49
 
50
+ def shot(brightness,image, labels_text):
51
  if "," in labels_text:
52
  prompts = labels_text.split(',')
53
  else:
 
55
  prompts = list(map(lambda x: x.strip(),prompts))
56
  predicted_images = detect_using_clip(image,prompts=prompts)
57
 
58
+ category_image = visualize_images(image=image,predicted_images=predicted_images,brightness=brightness)
59
  return category_image
60
 
61
  iface = gr.Interface(fn=shot,