Sophie98 commited on
Commit
7ae512f
1 Parent(s): f2e8e46

more testing

Browse files
Files changed (1) hide show
  1. segmentation.py +1 -1
segmentation.py CHANGED
@@ -47,7 +47,7 @@ def get_mask(image:Image) -> Image:
47
  # #load model
48
  # model.load_weights(model_path)
49
  model = keras.models.load_model('model_final.h5', compile=False)
50
-
51
  test_img = np.array(image)#cv2.imread(path, cv2.IMREAD_COLOR)
52
  test_img = cv2.resize(test_img, (640, 640))
53
  return test_img
 
47
  # #load model
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)#cv2.imread(path, cv2.IMREAD_COLOR)
52
  test_img = cv2.resize(test_img, (640, 640))
53
  return test_img