google-research / pix2seq

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

anybody knows how to train the custom datasets? #12

Open Isaiah1013 opened 2 years ago

mustafaozx commented 2 years ago

Hello, you can create or change CustomCoco TFDS. https://www.tensorflow.org/datasets/add_dataset