Spaces:
Build error
Build error
Sophie98
commited on
Commit
•
7ae512f
1
Parent(s):
f2e8e46
more testing
Browse files- 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
|