Spaces:
Runtime error
Runtime error
Commit
·
ab2f1c9
1
Parent(s):
858f61a
Delete make_predictions
Browse files- make_predictions +0 -20
make_predictions
DELETED
@@ -1,20 +0,0 @@
|
|
1 |
-
import torch
|
2 |
-
import numpy as np
|
3 |
-
import cv2
|
4 |
-
|
5 |
-
def make_predictions(model, input_img, threshold = 0.05):
|
6 |
-
model.eval()
|
7 |
-
with torch.no_grad():
|
8 |
-
image = input_img
|
9 |
-
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
10 |
-
image = image.astype("float32") / 255.0
|
11 |
-
image = cv2.resize(image, (height, width))
|
12 |
-
image = np.transpose(image, (2, 0, 1))
|
13 |
-
image = np.expand_dims(image, 0)
|
14 |
-
image = torch.from_numpy(image).to(device)
|
15 |
-
predMask = model(image).squeeze()
|
16 |
-
predMask = torch.sigmoid(predMask)
|
17 |
-
predMask = predMask.cpu().numpy()
|
18 |
-
predMask = (predMask > threshold) * 255
|
19 |
-
predMask = predMask.astype(np.uint8)
|
20 |
-
return predMask
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|