yolov11_heatmap.py 对分类不起作用

亲爱的 @z1069614715, 我试图在 yolo11-cls.pt 上使用您的脚本 yolov11_heatmap.py 进行分类任务。但是我收到了以下错误: 警告... self.method(tensor, [self.target]) 失败。 'tuple' 对象没有属性 'cpu'