BhumikaMak commited on
Commit
dae8a3a
·
verified ·
1 Parent(s): f79abc6

update results display

Browse files
Files changed (1) hide show
  1. yolov5.py +2 -2
yolov5.py CHANGED
@@ -90,10 +90,10 @@ def xai_yolov5(image,target_lyr = -5, n_components = 8):
90
 
91
  rgb_img_float, batch_explanations, result = dff_nmf(image, target_lyr = -5, n_components = 8)
92
  result = np.hstack(result)
93
- im = visualize_batch_explanations(rgb_img_float, batch_explanations)
94
 
95
  # Combine results
96
- final_image = np.hstack((image, detections_img, renormalized_cam_image, im))
97
  caption = "Results using YOLOv5"
98
  return Image.fromarray(final_image), caption, result
99
 
 
90
 
91
  rgb_img_float, batch_explanations, result = dff_nmf(image, target_lyr = -5, n_components = 8)
92
  result = np.hstack(result)
93
+ im = visualize_batch_explanations(rgb_img_float, batch_explanations) ##########to be displayed
94
 
95
  # Combine results
96
+ final_image = np.hstack((image, detections_img, renormalized_cam_image))
97
  caption = "Results using YOLOv5"
98
  return Image.fromarray(final_image), caption, result
99