debug: float range

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)
         
@@ -253,6 +253,7 @@ def dff_nmf(image, target_lyr, n_components):
 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
+        print("visualization concepts",rgb_img_float.shape,explanation.shape  )
         visualization = show_factorization_on_image(rgb_img_float, explanation, image_weight=image_weight)
         plt.figure()
         plt.imshow(visualization)  # Correctly pass the visualization data