Spaces:
Sleeping
Sleeping
Update inference_img.py
Browse files- inference_img.py +7 -7
inference_img.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
-
import os
|
2 |
-
import cv2
|
3 |
-
import torch
|
4 |
-
import argparse
|
5 |
-
from torch.nn import functional as F
|
6 |
-
import warnings
|
7 |
|
8 |
-
warnings.filterwarnings("ignore")
|
9 |
|
10 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
torch.set_grad_enabled(False)
|
|
|
1 |
+
import os
|
2 |
+
import cv2
|
3 |
+
import torch
|
4 |
+
import argparse
|
5 |
+
from torch.nn import functional as F
|
6 |
+
import warnings
|
7 |
|
8 |
+
warnings.filterwarnings("ignore")
|
9 |
|
10 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
torch.set_grad_enabled(False)
|