70e4855f5608c4481dfffd5f762e310d631d06c3,test_model_CAM.py,,returnCAM,#Any#Any#Any#,45

Before Change

    nc, h, w = feature_conv.shape
    output_cam = []
    for idx in class_idx:
        cam = weight_softmax[class_idx].dot(feature_conv.reshape((nc, h*w)))
        cam = cam.reshape(h, w)
        cam = cam - np.min(cam)
        cam_img = cam / np.max(cam)
        cam_img = np.uint8(255 * cam_img)
        output_cam.append(imresize(cam_img, size_upsample))
    return output_cam

After Change

    output_cam = []
    for idx in class_idx:
        cam = (weight_softmax[class_idx]@feature_conv.view(nc, h*w))
        cam.add_(-cam.min()).div_(cam.max()).mul_(255)
        cam = F.interpolate(cam.view(1,1,h,w), size=size_upsample, mode="bilinear",
                            align_corners=False).squeeze()
        cam = np.uint8(cam.numpy())
        output_cam.append(cam)

Instances