Spaces:
Runtime error
Runtime error
extra
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
|
|
| 83 |
cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
|
| 84 |
|
| 85 |
grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
|
| 86 |
-
img = cv2.resize(image, (
|
| 87 |
img = np.float32(img) / 255
|
| 88 |
cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
|
| 89 |
|
|
|
|
| 83 |
cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
|
| 84 |
|
| 85 |
grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
|
| 86 |
+
img = cv2.resize(image, (608, 608))
|
| 87 |
img = np.float32(img) / 255
|
| 88 |
cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
|
| 89 |
|