Skip to content
Snippets Groups Projects
Verified Commit 188b679a authored by Mélodie Boillet's avatar Mélodie Boillet
Browse files

Do not create a tensor from input_tensor

parent cca9ff57
No related branches found
No related tags found
No related merge requests found
......@@ -283,7 +283,7 @@ def process_image(
logger.debug("Image pre-processed.")
# Convert to tensor of size (batch_size, channel, height, width) with batch_size=1
input_tensor = torch.tensor(im_p).unsqueeze(0)
input_tensor = im_p.unsqueeze(0)
input_tensor = input_tensor.to(device)
input_sizes = [im_p.shape[1:]]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment