Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def inference(img, task, run_on):
|
|
62 |
torch.cuda.empty_cache()
|
63 |
|
64 |
open_cv_image = np.array(img)
|
65 |
-
img = cv2.cvtColor(
|
66 |
|
67 |
input_ = torch.from_numpy(img).float().div(255.).permute(2,0,1).unsqueeze(0).to(device)
|
68 |
|
|
|
62 |
torch.cuda.empty_cache()
|
63 |
|
64 |
open_cv_image = np.array(img)
|
65 |
+
img = cv2.cvtColor(open_cv_image, cv2.COLOR_BGR2RGB)
|
66 |
|
67 |
input_ = torch.from_numpy(img).float().div(255.).permute(2,0,1).unsqueeze(0).to(device)
|
68 |
|