saritha5 commited on
Commit
f5803bc
·
1 Parent(s): 8ac4e31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
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
- 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
 
 
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