机器学习:创建班级激活图

时间:2018-06-21 02:58:20

标签: tensorflow machine-learning conv-neural-network convolutional-neural-network pre-trained-model

我最近按照本教程训练了自己的图像分类器

https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/?utm_campaign=chrome_series_machinelearning_063016&utm_source=gdev&utm_medium=yt-desc#0

对于不知道允许重新训练MobileNet网络的用户,可以拥有自己的自定义图像/类别。我已经使用tensorflow在自己的图像上建立了成功的训练网络,但现在想走得更远。

我现在想为输入到模型中的图像生成一个类别激活图(CAM)。我读过,您需要您的卷积神经网络具有一个用于CAM的全局平均池层,并且正在考虑MoblieNet网络不具备该层。 是否可以从我已经训练过的网络中生成CAM,还是需要使用VGG16这样的其他网络来进行再培训?

如果可能,有人可以向我指出如何生成CAM的正确方向吗?否则,如果有人不能向我指出正确的方向,说明如何使用我自己的图像重新训练另一个网络,以允许CAM以及如何在这些网络上创建CAM?

很抱歉,这个问题含糊不清。我绝不是熟悉和/或受过计算机科学训练的人,但是有兴趣了解更多信息。您的帮助将不胜感激。如果需要,请与我联系,进一步询问我的问题。希望很快能收到您的来信。

谢谢!

0 个答案:

没有答案