github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

choose aspect ration dependent on landscape or portrait (#48)

Browse Source
This commit is contained in:
Anders Aaen Springborg
2020-11-03 03:10:22 +01:00
committed by GitHub
parent d24363f421
commit 2808411421
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/efficientnet.py
View File

@@ -127,7 +127,9 @@ if __name__ == "__main__":
else:
url = "https://raw.githubusercontent.com/karpathy/micrograd/master/puppy.jpg"
img = Image.open(io.BytesIO(fetch(url)))
aspect_ratio = img.size[0] / img.size[1]
aspect_ratio_landscape = img.size[0] / img.size[1]
aspect_ratio_portrait = img.size[1] / img.size[0]
aspect_ratio = max (aspect_ratio_landscape, aspect_ratio_portrait)
img = img.resize((int(224*max(aspect_ratio,1.0)), int(224*max(1.0/aspect_ratio,1.0))))
img = np.array(img)