Update app.py
Browse files
app.py
CHANGED
|
@@ -94,9 +94,9 @@ def postprocess(pred, img0_shape, img):
|
|
| 94 |
return results
|
| 95 |
|
| 96 |
def detect_objects(image_path):
|
| 97 |
-
|
| 98 |
-
img, img0 = preprocess_image(image_path)
|
| 99 |
-
|
| 100 |
pred = infer(model, img)
|
| 101 |
results = postprocess(pred, img0.shape, img)
|
| 102 |
return results
|
|
|
|
| 94 |
return results
|
| 95 |
|
| 96 |
def detect_objects(image_path):
|
| 97 |
+
dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
|
| 98 |
+
#img, img0 = preprocess_image(image_path)
|
| 99 |
+
img, img0 = preprocess_image(dicom_image)
|
| 100 |
pred = infer(model, img)
|
| 101 |
results = postprocess(pred, img0.shape, img)
|
| 102 |
return results
|