Spaces:
Sleeping
Sleeping
Commit
·
ad8ea78
1
Parent(s):
3906e17
Update app.py
Browse files
app.py
CHANGED
@@ -202,7 +202,7 @@ with gr.Blocks() as app:
|
|
202 |
x = plt.show()
|
203 |
|
204 |
|
205 |
-
def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.
|
206 |
transparency: float = 0.5) -> List[np.ndarray]:
|
207 |
with torch.no_grad():
|
208 |
transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
|
|
|
202 |
x = plt.show()
|
203 |
|
204 |
|
205 |
+
def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.6, show_cam: bool = False,
|
206 |
transparency: float = 0.5) -> List[np.ndarray]:
|
207 |
with torch.no_grad():
|
208 |
transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
|