Spaces:
Sleeping
Sleeping
Commit
·
3252151
1
Parent(s):
588684f
Update app.py
Browse files
app.py
CHANGED
@@ -211,7 +211,7 @@ with gr.Blocks() as app:
|
|
211 |
x = plt.show()
|
212 |
|
213 |
|
214 |
-
def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.
|
215 |
transparency: float = 0.5) -> List[np.ndarray]:
|
216 |
with torch.no_grad():
|
217 |
transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
|
|
|
211 |
x = plt.show()
|
212 |
|
213 |
|
214 |
+
def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.6, show_cam: bool = False,
|
215 |
transparency: float = 0.5) -> List[np.ndarray]:
|
216 |
with torch.no_grad():
|
217 |
transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
|