[Keras] Keras问题汇总Q&A

List:

  • 运行Python脚本时显示Badfile:File is not a zip file?
  • 如何查看Keras自动定义类别的编号? / 使用训练好的模型预测时,预测概率序列和Labels的对应关系?
  • Keras能否使用不同尺寸的图像作为输入(不缩放)?
  • 如何提取模型中某一层的输入并可视化?
  • Keras如何调用多GPU,以及多GPU下如何保存模型?



Q:运行Python脚本时显示Badfile:File is not a zip file?


Q:如何查看Keras自动定义类别的编号? / 使用训练好的模型预测时,预测概率序列和Labels的对应关系?

print(validation_generator.class_indices)

使用生成器的.class_indices方法即可获取模型默认的Labels序列。
参考:
Keras flow_from_directory class index
Keras 训练时不用将数据全部加入内存


Q:如何解决数据不均衡问题?
fit函数中调用class_weight,可以通过字典设置每个类别输入权重,比如:cw = {0: 1, 1: 25},类别序列可以使用.class_indices获取。


Q:Keras能否使用不同尺寸的图像作为输入(不缩放)?
可行,但不推荐:


Q:如何提取模型中某一层的输入并可视化?
参见:[DeepLearning]keras初体验之病斑分类


Q:Keras如何调用多GPU,以及多GPU下如何保存模型?
multi_gpu_model

keras.utils.multi_gpu_model(model, gpus)

将模型在多个GPU上复制

特别地,该函数用于单机多卡的数据并行支持,它按照下面的方式工作:
(1)将模型的输入分为多个子batch
(2)在每个设备上调用各自的模型,对各自的数据集运行
(3)将结果连接为一个大的batch(在CPU上)
例如,你的batch_size是64而gpus=2,则输入会被分为两个大小为32的子batch,在两个GPU上分别运行,通过连接后返回大小为64的结果。 该函数线性的增加了训练速度,最高支持8卡并行。

*该函数只能在tf后端下使用

参数如下:

  • model: Keras模型对象,为了避免OOM错误(内存不足),该模型应在CPU上构建,参考下面的例子。
  • gpus: 大或等于2的整数,要并行的GPU数目。
    该函数返回Keras模型对象,它看起来跟普通的keras模型一样,但实际上分布在多个GPU上。

例子:

import tensorflow as tf
from keras.applications import Xception
from keras.utils import multi_gpu_model
import numpy as np

num_samples = 1000
height = 224
width = 224
num_classes = 1000

# Instantiate the base model
# (here, we do it on CPU, which is optional).
with tf.device('/cpu:0'):
    model = Xception(weights=None,
                     input_shape=(height, width, 3),
                     classes=num_classes)

# Replicates the model on 8 GPUs.
# This assumes that your machine has 8 available GPUs.
parallel_model = multi_gpu_model(model, gpus=8)
parallel_model.compile(loss='categorical_crossentropy',
                       optimizer='rmsprop')

# Generate dummy data.
x = np.random.random((num_samples, height, width, 3))
y = np.random.random((num_samples, num_classes))

# This `fit` call will be distributed on 8 GPUs.
# Since the batch size is 256, each GPU will process 32 samples.
parallel_model.fit(x, y, epochs=20, batch_size=256)

但是在parallel_model.fit()结束后,使用代码parallel_model.save()保存却出现错误:

parallel_model.save('test.h5')
Traceback (most recent call last):

  File "<ipython-input-13-8d4461a4551e>", line 1, in <module>
    parallel_model.save('test.h5')

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/engine/topology.py", line 2556, in save
    save_model(self, filepath, overwrite, include_optimizer)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/models.py", line 107, in save_model
    'config': model.get_config()

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/engine/topology.py", line 2397, in get_config
    return copy.deepcopy(config)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 230, in _deepcopy_list
    y.append(deepcopy(a, memo))

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 237, in _deepcopy_tuple
    y.append(deepcopy(a, memo))

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 237, in _deepcopy_tuple
    y.append(deepcopy(a, memo))

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 190, in deepcopy
    y = _reconstruct(x, rv, 1, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 334, in _reconstruct
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 190, in deepcopy
    y = _reconstruct(x, rv, 1, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 334, in _reconstruct
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 264, in _deepcopy_method
    return type(x)(x.im_func, deepcopy(x.im_self, memo), x.im_class)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 190, in deepcopy
    y = _reconstruct(x, rv, 1, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 334, in _reconstruct
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 230, in _deepcopy_list
    y.append(deepcopy(a, memo))

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 190, in deepcopy
    y = _reconstruct(x, rv, 1, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 334, in _reconstruct
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 298, in _deepcopy_inst
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 190, in deepcopy
    y = _reconstruct(x, rv, 1, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 334, in _reconstruct
    state = deepcopy(state, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 163, in deepcopy
    y = copier(x, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 257, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)

  File "/home/dexter/anaconda2/envs/tensorflow/lib/python2.7/copy.py", line 182, in deepcopy
    rv = reductor(2)

TypeError: can't pickle thread.lock objects

这个问题困扰了我很久,最后在 keras-team/keras/issues#8446&issues#8253找到正解。
不过当时提问者报错为:

TypeError: can’t pickle module objects

与我的TypeError: can't pickle thread.lock objects大同小异,解决方法如下:


意思就是直接使用传入方法keras.utils.multi_gpu_model(model, gpus)中的model即可,而不要使用返回的parallel_model,即:

model.save('xxx.h5')
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容