From b279127db188612aad4779a0480263eb01fbed07 Mon Sep 17 00:00:00 2001
From: chenjian <chenjian26@baidu.com>
Date: Wed, 27 Jul 2022 19:35:26 +0800
Subject: [PATCH] Fix bug in merge_images function

---
 visualdl/utils/img_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/visualdl/utils/img_util.py b/visualdl/utils/img_util.py
index 3d338e7a..fe95e634 100644
--- a/visualdl/utils/img_util.py
+++ b/visualdl/utils/img_util.py
@@ -35,11 +35,11 @@ def padding_image(img, height, width):
 
 def merge_images(imgs, dataformats, scale=1.0, rows=-1):
     assert rows <= len(imgs), "rows should not greater than numbers of pictures"
-    channel = imgs[0].shape[2]
     # convert format of each image to `hwc`
     for i, img in enumerate(imgs):
         imgs[i] = convert_to_HWC(img, dataformats)
 
+    channel = imgs[0].shape[2]
     height = -1
     width = -1
 
-- 
GitLab