Update yolov5.py

yolov5.py

@@ -236,7 +236,7 @@ def dff_nmf(image, target_lyr, n_components):
         print("____________image_arrya", image_array.shape)
         image_resized = cv2.resize(image_array, (640, 640))
         rgba_channels = cv2.split(image_resized)
-        alpha_channel = rgba_channels[3] / 255.0
+        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)
         
@@ -245,11 +245,8 @@ def dff_nmf(image, target_lyr, n_components):
         results.append(overlay_img)
         plt.clf()  
         #return image_array
-
-
     return rgb_img_float, batch_explanations, results
 
-
 def visualize_batch_explanations(rgb_img_float, batch_explanations, image_weight=0.7):
     for i, explanation in enumerate(batch_explanations):
         # Create visualization for each explanation