Update yolov5.py

yolov5.py

@@ -238,7 +238,8 @@ 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)
+        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]) )
         #visualization = show_factorization_on_image(rgb_img_float, image_array.resize((rgb_img_float.shape)) , image_weight=0.3)