diff options
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/images.py b/modules/images.py index 90eca37a..b9589563 100644 --- a/modules/images.py +++ b/modules/images.py @@ -24,10 +24,13 @@ def image_grid(imgs, batch_size=1, rows=None): rows = opts.n_rows
elif opts.n_rows == 0:
rows = batch_size
- else:
+ elif opts.grid_prevent_empty_spots:
rows = math.floor(math.sqrt(len(imgs)))
while len(imgs) % rows != 0:
rows -= 1
+ else:
+ rows = math.sqrt(len(imgs))
+ rows = round(rows)
cols = math.ceil(len(imgs) / rows)
|