vobecant
commited on
Commit
·
b60b305
1
Parent(s):
b0beb39
Initial commit.
Browse files- segmenter_model/utils.py +1 -1
segmenter_model/utils.py
CHANGED
|
@@ -369,7 +369,7 @@ def inference(
|
|
| 369 |
|
| 370 |
# for im, im_metas in zip(ims, ims_metas):
|
| 371 |
for im in ims:
|
| 372 |
-
im = im.to('cuda')
|
| 373 |
if len(im.shape) == 3:
|
| 374 |
im = im.unsqueeze(0)
|
| 375 |
# im = resize(im, window_size)
|
|
|
|
| 369 |
|
| 370 |
# for im, im_metas in zip(ims, ims_metas):
|
| 371 |
for im in ims:
|
| 372 |
+
# im = im.to('cuda')
|
| 373 |
if len(im.shape) == 3:
|
| 374 |
im = im.unsqueeze(0)
|
| 375 |
# im = resize(im, window_size)
|