还能坚持 2021-04-20 15:53 采纳率: 0%
浏览 26

MXNet如何实现Grad-CAM

在网上找到了一份由Gloun实现的Grad-CAM,可是当将MXNet生成的模型参数文件.params和.json,用Gloun加载后,在反向传播时候,报错提示缺少softmax_label

    symbol, arg_params, aux_params = mx.model.load_checkpoint(load_model_prefix, epoch_idx)
    layers = symbol.get_internals()     # 得到所有的layers
    outputs = layers['amsoftmax_output']   # 选择输出层
    inputs = layers['data']             # 选择输入层
    network = gluon.nn.SymbolBlock(outputs, inputs)     # 使用gluon的接口将其封装成一个新的net 

- 已经知道了问题的原因,但是,如何在加载模型的时候,将softmax_label标签顺带输入进去?

  • 写回答

1条回答 默认 最新

  • 有问必答小助手 2021-04-21 17:37
    关注

    你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答

    本次提问扣除的有问必答次数,将会以问答VIP体验卡(1次有问必答机会、商城购买实体图书享受95折优惠)的形式为您补发到账户。

    ​​​​因为有问必答VIP体验卡有效期仅有1天,您在需要使用的时候【私信】联系我,我会为您补发。

    评论

报告相同问题?