Thouph
/

experimental_efficientnetv2_m_8035

Model card Files Files and versions Community

Thouph commited on Nov 12, 2023

Commit

a142643

•

1 Parent(s): 5c38f17

Upload inference.py

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -26,7 +26,7 @@ img = Image.open(image_path).convert('RGB')
 aspect_ratio = img.width / img.height
 new_height = math.sqrt(512 ** 2 / aspect_ratio)
 new_width = aspect_ratio * new_height
-img.resize((int(new_width), int(new_height)), Image.LANCZOS)
 tensor = transform(img).unsqueeze(0).to(device)
 with torch.no_grad():
     out = model(tensor)

 aspect_ratio = img.width / img.height
 new_height = math.sqrt(512 ** 2 / aspect_ratio)
 new_width = aspect_ratio * new_height
+img.thumbnail((int(new_width), int(new_height)), Image.LANCZOS)
 tensor = transform(img).unsqueeze(0).to(device)
 with torch.no_grad():
     out = model(tensor)