(images, size)
| 61 | |
| 62 | |
| 63 | def mergeImgs(images, size): |
| 64 | h, w = images.shape[1], images.shape[2] |
| 65 | imgs = np.zeros((h * size[0], w * size[1], 3)) |
| 66 | for idx, image in enumerate(images): |
| 67 | i = idx % size[1] |
| 68 | j = idx // size[1] |
| 69 | imgs[j * h:j * h + h, i * w:i * w + w, :] = image |
| 70 | |
| 71 | return imgs |
| 72 | |
| 73 | |
| 74 | if __name__ == '__main__': |