Fantasy-Studio / Paint-by-Example

Paint by Example: Exemplar-based Image Editing with Diffusion Models
https://arxiv.org/abs/2211.13227
Other
1.03k stars 93 forks source link

Why are bboxes processed only for first 500 images in each directory? #23

Closed shamitlal closed 1 year ago

shamitlal commented 1 year ago

Hi, Thanks for open sourcing the implementation. I was going through the read_bbox.py script, and saw that you only process the first 500 images in each directory (train_0, train_1,....) Link for code: https://github.com/Fantasy-Studio/Paint-by-Example/blob/main/scripts/read_bbox.py#L26

Is there a reason for this? Wouldn't this limit the amount of training data to 16*500 images only?

Thanks Shamit

Fantasy-Studio commented 1 year ago

Sorry, this is a bug left by my testing before making the code public, and it has been fixed.