update-heatmap

nlcfdwr · Feb 2, 2024 · 149f8e4 · 149f8e4
1 parent 0f3a260
commit 149f8e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolo-gradcam/yolov8_heatmap.py b/yolo-gradcam/yolov8_heatmap.py
@@ -205,7 +205,7 @@ def __call__(self, img_path, save_path):
 
 def get_params():
     params = {
-        'weight': 'runs/train/exp2/weights/best.pt',
+        'weight': 'runs/train/exp2/weights/best.pt', # 现在只需要指定权重即可,不需要指定cfg
         'device': 'cuda:0',
         'method': 'HiResCAM', # GradCAMPlusPlus, GradCAM, XGradCAM, EigenCAM, HiResCAM, LayerCAM, RandomCAM, EigenGradCAM
         'layer': [10, 12, 14, 16, 18],