Update ImageNode.py

shadowcz007 · Oct 21, 2024 · a47a56b · a47a56b
1 parent 3667b42
commit a47a56b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
@@ -2951,13 +2951,15 @@ def run(self,width,height,scale_option,image=None,average_color=['on'],fill_colo
                     im=tensor2pil(im)
 
                     im=im.convert('RGB')
+
+                    a_im,hex=get_average_color_image(im)
 
                     if average_color=='on':
                         fill_color=hex
-
-                    im=resize_image(im,scale_option,w,h,fill_color)
 
-                    a_im,hex=get_average_color_image(im)
+                    a_im=resize_image(a_im,scale_option,w,h,fill_color)
+
+                    im=resize_image(im,scale_option,w,h,fill_color)
 
                     im=pil2tensor(im)
                     imgs.append(im)