wangjin2000
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ def detect_objects(image_path):
|
|
74 |
return results, dicom_image, dicom_meta
|
75 |
|
76 |
def draw_bounding_boxes(img, results):
|
77 |
-
for (x1, y1, x2, y2
|
78 |
x1, y1, x2, y2, cls = map(int, [x1, y1, x2, y2, cls])
|
79 |
#conf = map(float, conf)
|
80 |
print("results in draw_bounding box:",[(x1, y1, x2, y2), conf, cls])
|
|
|
74 |
return results, dicom_image, dicom_meta
|
75 |
|
76 |
def draw_bounding_boxes(img, results):
|
77 |
+
for (x1, y1, x2, y2, conf, cls) in results:
|
78 |
x1, y1, x2, y2, cls = map(int, [x1, y1, x2, y2, cls])
|
79 |
#conf = map(float, conf)
|
80 |
print("results in draw_bounding box:",[(x1, y1, x2, y2), conf, cls])
|