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
|