AItool commited on
Commit
a43d977
·
verified ·
1 Parent(s): e56abbd

Update inference_img.py

Browse files
Files changed (1) hide show
  1. 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)