Sophie98 commited on
Commit
defa533
1 Parent(s): 758d201

more testing

Browse files
Files changed (1) hide show
  1. segmentation.py +1 -0
segmentation.py CHANGED
@@ -48,6 +48,7 @@ def get_mask(image:Image) -> Image:
48
  model.load_weights(model_path)
49
  #model = keras.models.load_model('model_final.h5', compile=False)
50
  print('loaded model')
 
51
  test_img = np.array(image)
52
  test_img = cv2.resize(test_img, (640, 640))
53
  test_img = cv2.cvtColor(test_img, cv2.COLOR_RGB2BGR)
48
  model.load_weights(model_path)
49
  #model = keras.models.load_model('model_final.h5', compile=False)
50
  print('loaded model')
51
+ return image
52
  test_img = np.array(image)
53
  test_img = cv2.resize(test_img, (640, 640))
54
  test_img = cv2.cvtColor(test_img, cv2.COLOR_RGB2BGR)