google-research / pix2seq

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)
Apache License 2.0
857 stars 71 forks source link

'ImageFont' object has no attribute 'getsize' #48

Open JJJYmmm opened 1 year ago

JJJYmmm commented 1 year ago

When I run /colab/pix2seq_inference_object_detection.ipynb, I meet an error as shown in the following figure.

image

It shows that the problem lies in line233 and line243 in /tasks/visualization/vis_utils.py.

display_str_heights = [font.getsize(ds)[1] for ds in display_str_list] # line 233 
text_width, text_height = font.getsize(display_str) # line 243

class PIL.ImageFont(PIL.version=10.0.0) don't have function getsize() directly, you should get the class member font first, then you can use function getsize(). so the solution is below.

display_str_heights = [font.font.getsize(ds)[1] for ds in display_str_list] # line 233 
text_width, text_height = font.font.getsize(display_str) # line 243

Then,you can see it works! image