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