raghakot / keras-vis

Neural network visualization toolkit for keras
https://raghakot.github.io/keras-vis
MIT License
2.97k stars 664 forks source link

Cannot see a bird #138

Closed Ume0128 closed 5 years ago

Ume0128 commented 5 years ago

` from vis.losses import ActivationMaximization from vis.regularizers import TotalVariation, LPNorm from vis.input_modifiers import Jitter from PIL import Image import imageio from vis.optimizer import Optimizer

from vis.callbacks import GifGenerator from keras.applications.vgg16 import VGG16

model = VGG16(weights='imagenet', include_top=True) print('Model loaded.')

layer_name = 'predictions' layer_dict = dict([(layer.name, layer) for layer in model.layers[1:]]) output_class = [20]

losses = [ (ActivationMaximization(layer_dict[layer_name], output_class), 0.1), (LPNorm(model.input), 6), (TotalVariation(model.input), 10) ] opt = Optimizer(model.input, losses) opt.minimize(max_iter=1000, verbose=True, input_modifiers=[Jitter()], callbacks=[GifGenerator('opt_progress')]) `

I write above code, and this program generates gif image. But, I cannot see bird in the gif image. Please tell me how to do.

(I rewrite /vis/utils/utils.py ) img = Image.fromarray(img.astype('uint8')) draw = ImageDraw.Draw(img) draw.text(position, text, fill=color, font=font)

2018-10-05 13 51 33