Update README.md
Browse files
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)
|