return a list of images

yolov5.py

@@ -89,7 +89,7 @@ def xai_yolov5(image,target_lyr = -5, n_components = 8):
     cam_image, renormalized_cam_image = generate_cam_image(model, target_layers, tensor, image, boxes)
 
     rgb_img_float, batch_explanations, result = dff_nmf(image, target_lyr = -5, n_components = 8)
-    result = np.hstack(result)
+    #result = np.hstack(result)
     im = visualize_batch_explanations(rgb_img_float, batch_explanations)  ##########to be displayed
     
     # Combine results