Spaces:
Running
on
Zero
Running
on
Zero
Martin Tomov
commited on
"insect"
Browse files
app.py
CHANGED
|
@@ -187,7 +187,7 @@ def detections_to_json(detections):
|
|
| 187 |
return detections_list
|
| 188 |
|
| 189 |
def process_image(image, include_json):
|
| 190 |
-
labels = [
|
| 191 |
original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
|
| 192 |
yellow_background_with_insects = create_yellow_background_with_insects(np.array(original_image), detections)
|
| 193 |
if include_json:
|
|
|
|
| 187 |
return detections_list
|
| 188 |
|
| 189 |
def process_image(image, include_json):
|
| 190 |
+
labels = ["insect"]
|
| 191 |
original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
|
| 192 |
yellow_background_with_insects = create_yellow_background_with_insects(np.array(original_image), detections)
|
| 193 |
if include_json:
|