google-research / pix2seq

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

Input Sequence Box Augmentation #30

Closed ShijieVVu closed 1 year ago

ShijieVVu commented 1 year ago

Did you include the source code for noised gt box and hallucinated box augmentation described in the paper? I can only find class augmentation's code. Thanks!

chentingpc commented 1 year ago

https://github.com/google-research/pix2seq/blob/f213f3a8eda5590b261343026f4a4c951da58638/data/data_utils.py#L326

ShijieVVu commented 1 year ago

Thanks for pointing out!