Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,9 +40,10 @@ def load_model(model_repo_id):
|
|
| 40 |
def predict(pilimg,threshold):
|
| 41 |
|
| 42 |
image_np = pil_image_as_numpy_array(pilimg)
|
| 43 |
-
|
|
|
|
| 44 |
|
| 45 |
-
def predict2(image_np,threshold):
|
| 46 |
|
| 47 |
results = detection_model(image_np)
|
| 48 |
|
|
@@ -59,7 +60,7 @@ def predict2(image_np,threshold):
|
|
| 59 |
result['detection_scores'][0],
|
| 60 |
category_index,
|
| 61 |
use_normalized_coordinates=True,
|
| 62 |
-
max_boxes_to_draw=
|
| 63 |
min_score_thresh=threshold,
|
| 64 |
agnostic_mode=False,
|
| 65 |
line_thickness=2)
|
|
|
|
| 40 |
def predict(pilimg,threshold):
|
| 41 |
|
| 42 |
image_np = pil_image_as_numpy_array(pilimg)
|
| 43 |
+
# threshold1=threshold
|
| 44 |
+
# return predict2(image_np,threshold)
|
| 45 |
|
| 46 |
+
# def predict2(image_np,threshold):
|
| 47 |
|
| 48 |
results = detection_model(image_np)
|
| 49 |
|
|
|
|
| 60 |
result['detection_scores'][0],
|
| 61 |
category_index,
|
| 62 |
use_normalized_coordinates=True,
|
| 63 |
+
max_boxes_to_draw=20,
|
| 64 |
min_score_thresh=threshold,
|
| 65 |
agnostic_mode=False,
|
| 66 |
line_thickness=2)
|