from
PIL
import
Image, ImageDraw
import
numpy as np
def
square_thumb(thum_img,width,height):
if
height
=
=
width:
return
thum_img
elif
height > width:
square_img
=
Image.new(thum_img.mode, (height, height))
square_img.paste(thum_img, ((height
-
width)
/
/
2
,
0
))
return
square_img
else
:
square_img
=
Image.new(thum_img.mode, (width, width))
square_img.paste(thum_img, (
0
, (width
-
height)
/
/
2
))
return
square_img
if
__name__
=
=
"__main__"
:
orig_img
=
Image.
open
(
'sebastian-molina.jpg'
)
w,h
=
orig_img.size
sq_img
=
square_thumb(orig_img,w,h)
sq_img.thumbnail((
400
,
400
))
sq_img.save(
'Square_thumbnail.jpg'
)