Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,19 +19,19 @@ model = core.Model.load(MODEL_PATH, ['cross_arm','pole','tag'])
|
|
| 19 |
st.title("Object Detection")
|
| 20 |
|
| 21 |
|
| 22 |
-
def detect_object(IMAGE_PATH):
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
thresh=0.2
|
| 29 |
-
filtered_indices=np.where(scores>thresh)
|
| 30 |
-
filtered_scores=scores[filtered_indices]
|
| 31 |
-
filtered_boxes=boxes[filtered_indices]
|
| 32 |
-
num_list = filtered_indices[0].tolist()
|
| 33 |
-
filtered_labels = [labels[i] for i in num_list]
|
| 34 |
-
visualize.show_labeled_image(image, filtered_boxes, filtered_labels)
|
| 35 |
#img_array = img_to_array(img)
|
| 36 |
|
| 37 |
|
|
|
|
| 19 |
st.title("Object Detection")
|
| 20 |
|
| 21 |
|
| 22 |
+
#def detect_object(IMAGE_PATH):
|
| 23 |
+
# image = utils.read_image(IMAGE_PATH)
|
| 24 |
+
# predictions = model.predict(image)
|
| 25 |
+
# labels, boxes, scores = predictions
|
| 26 |
+
|
| 27 |
+
|
| 28 |
+
#thresh=0.2
|
| 29 |
+
#filtered_indices=np.where(scores>thresh)
|
| 30 |
+
#filtered_scores=scores[filtered_indices]
|
| 31 |
+
#filtered_boxes=boxes[filtered_indices]
|
| 32 |
+
#num_list = filtered_indices[0].tolist()
|
| 33 |
+
#filtered_labels = [labels[i] for i in num_list]
|
| 34 |
+
#visualize.show_labeled_image(image, filtered_boxes, filtered_labels)
|
| 35 |
#img_array = img_to_array(img)
|
| 36 |
|
| 37 |
|