AkashDataScience commited on
Commit
655a5f6
·
1 Parent(s): 87700f1

Updated default values

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -52,6 +52,7 @@ def display_false_detection_data(false_detection_data, number_of_samples):
52
  for i in range(number_of_samples):
53
  plt.subplot(y_count, x_count, i + 1)
54
  img = cv2.imread(false_detection_data[i])
 
55
  plt.imshow(img)
56
  plt.xticks([])
57
  plt.yticks([])
@@ -110,11 +111,16 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
110
 
111
  title = "YOLOv9 model to detect shirt/tshirt"
112
  description = "A simple Gradio interface to infer on YOLOv9 model and detect tshirt in image"
113
- examples = [["image_1.jpg", 0.25, 0.45], ["image_2.jpg", 0.25, 0.45],
114
- ["image_3.jpg", 0.25, 0.45], ["image_4.jpg", 0.25, 0.45],
115
- ["image_5.jpg", 0.25, 0.45], ["image_6.jpg", 0.25, 0.45],
116
- ["image_7.jpg", 0.25, 0.45], ["image_8.jpg", 0.25, 0.45],
117
- ["image_9.jpg", 0.25, 0.45], ["image_10.jpg", 0.25, 0.45]]
 
 
 
 
 
118
 
119
  demo = gr.Interface(inference,
120
  inputs = [gr.Image(width=320, height=320, label="Input Image"),
 
52
  for i in range(number_of_samples):
53
  plt.subplot(y_count, x_count, i + 1)
54
  img = cv2.imread(false_detection_data[i])
55
+ img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
56
  plt.imshow(img)
57
  plt.xticks([])
58
  plt.yticks([])
 
111
 
112
  title = "YOLOv9 model to detect shirt/tshirt"
113
  description = "A simple Gradio interface to infer on YOLOv9 model and detect tshirt in image"
114
+ examples = [["image_1.jpg", 0.25, 0.45, True, 10],
115
+ ["image_2.jpg", 0.25, 0.45, True, 10],
116
+ ["image_3.jpg", 0.25, 0.45, True, 10],
117
+ ["image_4.jpg", 0.25, 0.45, True, 10],
118
+ ["image_5.jpg", 0.25, 0.45, True, 10],
119
+ ["image_6.jpg", 0.25, 0.45, True, 10],
120
+ ["image_7.jpg", 0.25, 0.45, True, 10],
121
+ ["image_8.jpg", 0.25, 0.45, True, 10],
122
+ ["image_9.jpg", 0.25, 0.45, True, 10],
123
+ ["image_10.jpg", 0.25, 0.45, True, 10]]
124
 
125
  demo = gr.Interface(inference,
126
  inputs = [gr.Image(width=320, height=320, label="Input Image"),