8wDlpd.png
8wDFp9.png
8wDEOx.png
8wDMfH.png
8wDKte.png

TimeDistributed.call() 接收的参数:

Christian Vincenzo Traina 2月前

120 0

我正在尝试做一个视频分类 dl 项目。预处理:从每个视频中,我取出 20 帧并重新整形它们并将它们添加到一个列表中,我的 x.shape 看起来像这样 --> (无,20,2...

我正在尝试做一个视频分类 dl 项目。预处理:从每个视频中,我取出 20 帧并重新整形它们并将它们添加到列表中,我的 x.shape 看起来像这样 --> (None, 20, 224, 224, 3)模型:我使用 vgg16 作为基础模型并组合使用 lstm 层。baseModel = VGG16(weights=\'imagenet\', include_top=False, input_shape=(frame_height, frame_width, 3))

#model
model = Sequential()
model.add(TimeDistributed(baseModel, input_shape=(sequence, frame_height, frame_width, 3)))
model.add(TimeDistributed(Dropout(0.3)))
model.add(TimeDistributed(Flatten()))
model.add(TimeDistributed(Dropout(0.3)))
#LSTM Layer
model.add(LSTM(320, activation='tanh'))
model.add(Dropout(0.2))
model.add(Dense(label_count, activation="softmax")) 

for layer in baseModel.layers:
    layer.trainable = False

optimizers = Adam(learning_rate=0.001)
model.compile(loss='sparse_categorical_crossentropy', optimizer=optimizers, metrics=['accuracy'])

model.fit(x=x_train, y=y_train, validation_data=(x_val,y_val),epochs=1, batch_size=10, callbacks=[early_stopping])

此后,我将模型保存在本地:model.save(\'model.keras\')

但是,当我尝试使用 :loaded_model = load_model(r'model.keras') 加载模型时

我收到一个 ValueError:无法将“20”转换为形状。

TimeDistributed.call() 接收的参数:• args=(' ',)• kwargs={'mask': 'None'}

我想保存模型以供以后使用并在需要时加载它。我很新手,所以任何事情都会有帮助,即使有更好的预处理方法或更好的模型架构。

帖子版权声明 1、本帖标题:TimeDistributed.call() 接收的参数:
    本站网址:http://xjnalaquan.com/
2、本网站的资源部分来源于网络,如有侵权,请联系站长进行删除处理。
3、会员发帖仅代表会员个人观点,并不代表本站赞同其观点和对其真实性负责。
4、本站一律禁止以任何方式发布或转载任何违法的相关信息,访客发现请向站长举报
5、站长邮箱:yeweds@126.com 除非注明,本帖由Christian Vincenzo Traina在本站《tensorflow》版块原创发布, 转载请注明出处!
最新回复 (0)
  • 多个对象上的脚本默认是一个新实例,因此其中任何非类变量的变量对于每个项目都是唯一的,因此健康状况是针对每个敌人的,而不是针对所有人的。

返回
作者最近主题: