search

google-research / pix2seq

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

Made it easier to play with the inference demo - choosing different models and coco images via dropdown #8

Closed Nyandwi closed 2 years ago

Nyandwi commented 2 years ago

I was playing with the Pix2seq inference demo and thought it would be nice to be able to try different models and COCO validation images by choosing them via dropdown.

Thus, I added all pre-trained models and original COCO images from the paper into dropdown form (in section G VISUALIZATION OF DETECTION RESULTS) to make it easy to try different models & test images.

If this would be useful for people that are playing with the notebooks, I would also be happy to do the same thing in fine-tuning notebook.

Thanks!

chentingpc commented 2 years ago

This looks great! Thanks for your contribution.

Nyandwi commented 2 years ago

Thank you too! Pix2seq is really an amazing work.