Spaces:
Sleeping
Sleeping
update results display
Browse files
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
|
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 |
|