Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def get_mask(model, input_img, use_amp=True, s=640):
|
|
40 |
def process(img):
|
41 |
path = load_img(img,output_type="str")
|
42 |
img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
|
43 |
-
mask = get_mask(
|
44 |
img = mask * img
|
45 |
out = load_img(img)
|
46 |
return out
|
|
|
40 |
def process(img):
|
41 |
path = load_img(img,output_type="str")
|
42 |
img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
|
43 |
+
mask = get_mask(model, img, use_amp= False, s=img_size)
|
44 |
img = mask * img
|
45 |
out = load_img(img)
|
46 |
return out
|