Merge pull request #7344 from glop102/master
Reduce grid rows if larger than number of images available
This commit is contained in:
commit
1d24665229
|
@ -36,6 +36,8 @@ def image_grid(imgs, batch_size=1, rows=None):
|
||||||
else:
|
else:
|
||||||
rows = math.sqrt(len(imgs))
|
rows = math.sqrt(len(imgs))
|
||||||
rows = round(rows)
|
rows = round(rows)
|
||||||
|
if rows > len(imgs):
|
||||||
|
rows = len(imgs)
|
||||||
|
|
||||||
cols = math.ceil(len(imgs) / rows)
|
cols = math.ceil(len(imgs) / rows)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue