OriLib commited on
Commit
398a3d9
1 Parent(s): aab1fdd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -4
README.md CHANGED
@@ -116,10 +116,7 @@ net.eval()
116
  model_input_size = [1024,1024]
117
  orig_im = io.imread(im_path)
118
  orig_im_size = orig_im.shape[0:2]
119
- image = preprocess_image(orig_im, model_input_size)
120
-
121
- if torch.cuda.is_available():
122
- image=image.cuda()
123
 
124
  # inference
125
  result=net(image)
 
116
  model_input_size = [1024,1024]
117
  orig_im = io.imread(im_path)
118
  orig_im_size = orig_im.shape[0:2]
119
+ image = preprocess_image(orig_im, model_input_size).to(device)
 
 
 
120
 
121
  # inference
122
  result=net(image)