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 |
|