Update yolov5.py

yolov5.py

@@ -232,7 +232,7 @@ def dff_nmf(image, target_lyr, n_components):
 
         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[(image_array.shape[0]//2) :, :])
+        results.append(image_array)
         plt.clf()  
         #return image_array