updated overlay visualisation

yolov5.py

@@ -238,11 +238,12 @@ def dff_nmf(image, target_lyr, n_components):
         rgba_channels = cv2.split(image_resized)
         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)
+        #overlay_img = (alpha_channel[..., None] * image) + ((1 - alpha_channel[..., None]) * rgb_channels)
         
         #temp = image_array.reshape((rgb_img_float.shape[0],rgb_img_float.shape[1]) )
         #visualization = show_factorization_on_image(rgb_img_float, image_array.resize((rgb_img_float.shape)) , image_weight=0.3)
-        results.append(overlay_img)
+        visualization = show_factorization_on_image(rgb_img_float, np.transpose(rgb_channels, (2, 0, 1)), image_weight=0.3)
+        results.append(visualization)
         plt.clf()  
         #return image_array