wangjin2000 commited on
Commit
ae0e58e
·
verified ·
1 Parent(s): f866ff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -97,9 +97,9 @@ def postprocess(pred, img0_shape, img):
97
  return results
98
 
99
  def detect_objects(image_path):
100
- dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
101
- #img, img0 = preprocess_image(image_path)
102
- img, img0 = preprocess_image(dicom_image)
103
  pred = infer(model, img)
104
  results = postprocess(pred, img0.shape, img)
105
  return results
 
97
  return results
98
 
99
  def detect_objects(image_path):
100
+ #dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
101
+ img, img0 = preprocess_image(image_path)
102
+ #img, img0 = preprocess_image(dicom_image)
103
  pred = infer(model, img)
104
  results = postprocess(pred, img0.shape, img)
105
  return results