debug: shape broadcast

yolov5.py

@@ -239,6 +239,7 @@ def dff_nmf(image, target_lyr, n_components):
         alpha_channel = rgba_channels[3] 
         rgb_channels = np.stack(rgba_channels[:3], axis=-1)
         #overlay_img = (alpha_channel[..., None] * image) + ((1 - alpha_channel[..., None]) * rgb_channels)
+        print("shape....", rgb_img_float.shape, rgb_channels.shape)
         visualization = show_factorization_on_image(rgb_img_float, rgb_channels , image_weight=0.3)
         
         #temp = image_array.reshape((rgb_img_float.shape[0],rgb_img_float.shape[1]) )