update: direct array return

yolov5.py

@@ -228,10 +228,15 @@ def dff_nmf(image, target_lyr, n_components):
         predicted_label = labels[top_class_id]  # Map ID to label
         ax.text(x1, y1, f"{predicted_label}: {top_score:.2f}",
             color='r', fontsize=12, verticalalignment='top')
-        plt.show()
-        plt.savefig("test_" + str(indx) + ".png" )
-        plt.clf()
-        results.append(Image.open(f"test_{indx}.png"))
+       
+
+        fig.canvas.draw()  # Draw the canvas to make sure the image is rendered
+        image_array = np.array(fig.canvas.renderer.buffer_rgba())  # Convert to numpy array
+        results.append(image_array)
+        plt.clf()  
+
+        return image_array
+
 
     return rgb_img_float, batch_explanations, results