search

dailenson / One-DM

Official Code for ECCV 2024 paper — One-Shot Diffusion Mimicker for Handwritten Text Generation
MIT License
298 stars 28 forks source link

Random_StyleIAMDataset bug? #18

Open vincentkk6356 opened 4 weeks ago

vincentkk6356 commented 4 weeks ago

https://github.com/dailenson/One-DM/blob/bc705c0842a13ffabd8deaeda8bfef97bbbd50b4/data_loader/loader.py#L203-L217

Why is this function looping all the images instead of just picking one? It slows down the inference by quite a lot.

dailenson commented 3 weeks ago

This function isn't looping through all the images. It stops once it finds an image with a width greater than 128 pixels. The reasoning behind this is that longer images help ensure more stable performance, as mentioned in Section I of our paper.