Update app.py

app.py

@@ -73,13 +73,17 @@ def detect_objects(image_path):
     #return results, dicom_image
     return results, dicom_image, dicom_meta
 
-def draw_bounding_boxes(img, results):
+def draw_bounding_boxes(img, results, dicom_meta):
     dets = []
     for (x1, y1, x2, y2), conf, cls in results:
-        x1, y1, x2, y2, cls = map(int, [x1, y1, x2, y2, cls])
+        xc = x1+(x2-x1)/2
+        yc = y1+(y2-y1)/2
+        x1, y1, x2, y2, xc, yc, cls = map(int, [x1, y1, x2, y2, xc, yc, cls])
         conf = round(conf.detach().item(), 4)
         print("results in draw_bounding box:",[(x1, y1, x2, y2), conf, cls])
-        dets.append([(x1, y1, x2, y2), conf, cls])
+        zc = dicom_meta.loc[dicom_meta.Key == 'Instance Number', 'Value'].iloc[0]
+        #dets.append([(x1, y1, x2, y2), conf, cls])
+        dets.append([(xc, yc, zc), conf, cls])
         cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), 2)
         cv2.putText(img, f'{model.names[int(cls)]} {conf:.2f}', (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (36, 255, 12), 2)
     return img, dets
@@ -87,7 +91,7 @@ def draw_bounding_boxes(img, results):
 def show_preds_image(filepath):
     results, img0, dicom_meta = detect_objects(filepath)
     img_with_boxes, results = draw_bounding_boxes(img0, results)
-    print("In show_preds_image:Intance Number=", dicom_meta.loc[dicom_meta.Key == 'Instance Number', 'Value'].iloc[0])
+    #print("In show_preds_image:Intance Number=", dicom_meta.loc[dicom_meta.Key == 'Instance Number', 'Value'].iloc[0])
     #return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB)
     return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB), results, dicom_meta