Update app.py
Browse files
app.py
CHANGED
@@ -35,8 +35,11 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
|
|
35 |
for r in results:
|
36 |
conf = r.boxes.conf
|
37 |
cls = r.boxes.cls
|
|
|
|
|
|
|
38 |
|
39 |
-
text += (f"Detected {label[int(cls)]
|
40 |
|
41 |
return render,text
|
42 |
|
|
|
35 |
for r in results:
|
36 |
conf = r.boxes.conf
|
37 |
cls = r.boxes.cls
|
38 |
+
xywh = int(r.boxes.xywh)
|
39 |
+
x = xywh[0]
|
40 |
+
y = xywh[1]
|
41 |
|
42 |
+
text += (f"Detected {label[int(cls)]} with confidence {double(conf)} at {x}, {y}\n")
|
43 |
|
44 |
return render,text
|
45 |
|