Spaces:
Sleeping
Sleeping
Commit
·
655a5f6
1
Parent(s):
87700f1
Updated default values
Browse files
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
|
114 |
-
["
|
115 |
-
["
|
116 |
-
["
|
117 |
-
["
|
|
|
|
|
|
|
|
|
|
|
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"),
|