test
Browse files
app.py
CHANGED
@@ -64,9 +64,8 @@ def predict(inp):
|
|
64 |
# eigen_smooth=args.eigen_smooth,
|
65 |
# aug_smooth=args.aug_smooth
|
66 |
)
|
67 |
-
|
68 |
-
|
69 |
-
print(img.shape, img.dtype)
|
70 |
print(grad.shape, grad.dtype)
|
71 |
cam_image = show_cam_on_image(rgb_img, grad)
|
72 |
return confidences,cam_image
|
|
|
64 |
# eigen_smooth=args.eigen_smooth,
|
65 |
# aug_smooth=args.aug_smooth
|
66 |
)
|
67 |
+
grad = grad[0, :]
|
68 |
+
print(rgb_img.shape, rgb_img.dtype)
|
|
|
69 |
print(grad.shape, grad.dtype)
|
70 |
cam_image = show_cam_on_image(rgb_img, grad)
|
71 |
return confidences,cam_image
|