xiazhi commited on
Commit
4e59632
·
verified ·
1 Parent(s): ede6aa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -59,6 +59,8 @@ def click_process(image_input , type_dataset):
59
  if type_dataset == "pannuke":
60
  pannuke_inf.run_single_image_inference(pannuke_model,image_input)
61
  else:
 
 
62
  monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
63
 
64
  image_output = cv2.imread("pred_img.png")
 
59
  if type_dataset == "pannuke":
60
  pannuke_inf.run_single_image_inference(pannuke_model,image_input)
61
  else:
62
+ resize_shape = (512,512)
63
+ image_input = cv2.resize(image_input, resize_shape)
64
  monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
65
 
66
  image_output = cv2.imread("pred_img.png")