Update app.py

app.py

@@ -86,12 +86,14 @@ def scale_coords(img1_shape, coords, img0_shape, ratio_pad=None):
     coords[:, :4].clip_(min=0, max=img1_shape[0])  # clip boxes
     return coords
 
-def postprocess(pred, img0_shape, img):
+#def postprocess(pred, img0_shape, img):
+def postprocess(pred, img0, img):    
     pred = non_max_suppression(pred, conf_thres=0.25, iou_thres=0.45, classes=None, agnostic=False)
     results = []
     for det in pred:  # detections per image
         if len(det):
-            det[:, :4] = scale_coords(img.shape[2:], det[:, :4], img0_shape).round()
+            #det[:, :4] = scale_coords(img.shape[2:], det[:, :4], img0_shape).round()
+            det[:, :4] = scale_coords(img.shape[2:], det[:, :4], img0.shape).round()
             for *xyxy, conf, cls in reversed(det):
                 results.append((xyxy, conf, cls))
     return results
@@ -101,7 +103,8 @@ def detect_objects(image_path):
     #img, img0 = preprocess_image(image_path)
     img, img0 = preprocess_image(dicom_image)
     pred = infer(model, img)
-    results = postprocess(pred, img0.shape, img)
+    #results = postprocess(pred, img0.shape, img)
+    results = postprocess(pred, dicom_image, img)
     return results
 
 def draw_bounding_boxes(img, results):
@@ -112,8 +115,10 @@ def draw_bounding_boxes(img, results):
     return img
 
 def show_preds_image(filepath):
+    dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
     results = detect_objects(filepath)
-    img0 = cv2.imread(filepath)
+    #img0 = cv2.imread(filepath)
+    #img_with_boxes = draw_bounding_boxes(img0, results)
     img_with_boxes = draw_bounding_boxes(img0, results)
     return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB)