Spaces:
Running
Running
Commit
·
183692d
1
Parent(s):
bba5a32
adding app with CLIP image segmentation
Browse files
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=
|
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,
|