Skip to content

Commit

Permalink
Whoops forgot to include this in the last one
Browse files Browse the repository at this point in the history
  • Loading branch information
dbolya committed May 11, 2019
1 parent 8fed726 commit a70b68d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -558,7 +558,7 @@ def badhash(x):
return x

def evalimage(net:Yolact, path:str, save_path:str=None):
frame = torch.Tensor(cv2.imread(path)).cuda().float()
frame = torch.from_numpy(cv2.imread(path)).cuda().float()
batch = FastBaseTransform()(frame.unsqueeze(0))
preds = net(batch)

Expand Down

0 comments on commit a70b68d

Please sign in to comment.