Spaces:
Runtime error
Runtime error
extra
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
|
|
| 68 |
transformed_image = transforms(image=image)["image"].unsqueeze(0)
|
| 69 |
print("transformed_image 2")
|
| 70 |
output = model(transformed_image)
|
| 71 |
-
|
| 72 |
bboxes = [[] for _ in range(1)]
|
| 73 |
|
| 74 |
print("transformed_image 4")
|
|
@@ -88,7 +88,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
|
|
| 88 |
|
| 89 |
print("transformed_image 6")
|
| 90 |
grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
|
| 91 |
-
img = cv2.resize(image, (
|
| 92 |
img = np.float32(img) / 255
|
| 93 |
cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
|
| 94 |
|
|
|
|
| 68 |
transformed_image = transforms(image=image)["image"].unsqueeze(0)
|
| 69 |
print("transformed_image 2")
|
| 70 |
output = model(transformed_image)
|
| 71 |
+
print("transformed_image 3")
|
| 72 |
bboxes = [[] for _ in range(1)]
|
| 73 |
|
| 74 |
print("transformed_image 4")
|
|
|
|
| 88 |
|
| 89 |
print("transformed_image 6")
|
| 90 |
grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
|
| 91 |
+
img = cv2.resize(image, (608, 608))
|
| 92 |
img = np.float32(img) / 255
|
| 93 |
cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
|
| 94 |
|