关于旋转船舶热力图的生成操作概略过程

本文记录了如何重新熟悉并使用热力图生成程序,特别是针对旋转框IOU计算的挑战。在Grad-CAM的实现中,详细介绍了groundtruth bounding box的处理,以及在demo_grad_cam中配置文件和图像名称的设定。运行代码需在终端指定参数,注意得分和回归热力图的反向传播选择。
摘要由CSDN通过智能技术生成

好久之后再次使用生成热力图的程序,居然忘记了怎么使用,摸索了一会才搞通。所以这次就简单记录一下使用过程,方便以后快速使用。

代码在之前都已经编好,可是借鉴的代码都忘了在哪了,只记得百度热力图的生成,找到一些自然图像的生成方式,分类得分的热力图还好说,但是回归的热力图因为涉及到旋转框IOU的计算,又找了一些旋转IOU计算的方法,网上不是所有的计算方法都可以,因为这里涉及到tensor的反向传播,后来找到了一个方法可使用。

具体的应用是在grad_cam中的__call__方法中,有一行代码是写明ground truth bounding box的,用来计算IOU。

在demo_grad_cam中,get_parser方法中指明了配置文件,图像名称,在save_image方法中指明了存储路径和存储的文件名。

运行文件还是需要在terminal中写运行代码,还要写明参数文件 --opts MODEL.WEIGHTS。不能直接用pycharm中的运行符号进行运行。

得分和回归的热力图需要在grad_cam的__call__中,对于得分还是IOU进行反向传播需要二选一。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值