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