f92dd441311594f80ee5be4a2d2a1d6b21b96c59,examples/visualize_layer.py,,visualize_multiple_categories,#Any#,69

Before Change

    // Visualize [20] (ouzel) and [20, 71] (An ouzel-scorpion :D)
    indices = [20, [20, 71]]
    images = [visualize_activation(model, layer_idx, filter_indices=idx,
                                   text=utils.get_imagenet_label(idx),
                                   max_iter=500) for idx in indices]

    // Easily stitch images via `utils.stitch_images`
    stitched = utils.stitch_images(images)
    if show:

After Change

    // Visualize [20] (ouzel) and [20, 71] (An ouzel-scorpion :D)
    indices = [20, [20, 71]]
    images = []
    for idx in indices:
        img = visualize_activation(model, layer_idx, filter_indices=idx, max_iter=500)
        img = utils.draw_text(img, utils.get_imagenet_label(idx))
        images.append(img)

    // Easily stitch images via `utils.stitch_images`
    stitched = utils.stitch_images(images)
    if show:
        plt.axis("off")
        plt.imshow(stitched)

Instances