UPDATE: DOC STRING

Spritan · Feb 27, 2024 · a12dd5c · a12dd5c
1 parent 7437226
commit a12dd5c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/YOLOv8_Explainer/core.py b/YOLOv8_Explainer/core.py
@@ -152,7 +152,6 @@ class yolov8_heatmap:
             device (str): The device to use for inference. Defaults to "cuda:0" if a GPU is available, otherwise "cpu".
             method (str): The method to use for computing the CAM. Defaults to "EigenCAM".
             layer (list): The indices of the layers to use for computing the CAM. Defaults to [10, 12, 14, 16, 18, -3].
-            backward_type (str): The type of backward pass to use. Can be "all", "positive", or "negative". Defaults to "all".
             conf_threshold (float): The confidence threshold for detections. Defaults to 0.2.
             ratio (float): The ratio of maximum scores to return. Defaults to 0.02.
             show_box (bool): Whether to show bounding boxes with the CAM. Defaults to True.