首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何根据提取的特征构造视频数据集,以构建CNN-LSTM分类模型?

如何根据提取的特征构造视频数据集,以构建CNN-LSTM分类模型?
EN

Stack Overflow用户
提问于 2020-04-09 23:32:49
回答 1查看 56关注 0票数 0

对于我的处理情感识别的项目,我有一个由多个视频组成的数据集,范围从0.5s到10s。我有一个应用程序,它遍历每个视频,并创建一个.csv文件,其中包含它从视频中的每一帧提取的特征,即,每行代表视频中的每一帧(没有。行是可变的),列表示应用程序从框架中提取的不同特征(没有。固定列的数量)。每个.csv文件名还包含表示视频中表达的情感的代码。

最初,我的计划是从视频中提取每一帧,并将每一帧作为输入传递到我计划使用的以下CNN-LSTM (CNN用于空间特征,LSTM用于时间特征)模型。

代码语言:javascript
复制
    model = Sequential()

    model.add(Input(input_shape))

    model.add(Conv3D(6, (1, 5, 5), (1, 1, 1), activation='relu', name='conv-1'))
    model.add(AveragePooling3D((1, 2, 2), strides=(1, 2, 2), name='avgpool-1'))

    model.add(Conv3D(16, (1, 5, 5), (1, 1, 1), activation='relu', name='conv-2'))
    model.add(AveragePooling3D((1, 2, 2), strides=(1, 2, 2), name='avgpool-2'))

    model.add(Conv3D(32, (1, 5, 5), (1, 1, 1), activation='relu', name='conv-3'))
    model.add(AveragePooling3D((1, 2, 2), strides=(1, 2, 2), name='avgpool-3'))

    model.add(Conv3D(64, (1, 4, 4), (1, 1, 1), activation='relu', name='conv-4'))
    model.add(Reshape((30, 64), name='reshape'))

    model.add(CuDNNLSTM(64, return_sequences=True, name='lstm-1'))
    model.add(CuDNNLSTM(64, name='lstm-2'))

    model.add(Dense(6, activation=tf.nn.softmax, name='result')) 

我仍然计划使用CNN-LSTM模型,但我现在不知道如何组织我的数据集。我想用相应的情感标签标记每个.csv文件中的每一帧,然后将所有.csv文件合并成一个.csv文件。在改变了输入形状和其他必要的参数后,这个组合的.csv文件将被传递给上面的模型,但我不知道如果这样做,模型是否能够区分视频。

因此,最后,我需要帮助构建我的数据集,以及如何将此数据集传递给CNN-LSTM模型。

EN

回答 1

Stack Overflow用户

发布于 2020-04-20 01:23:05

通过查看您的问题陈述,我认为没有必要区分视频。

您可以继续使用您的方法来标记视频中的每一帧,并将其合并为单个CSV文件。

For可以使用下面的代码从CSV文件转换为NumPy数组,以按照下面的方法准备模型进行训练。

代码语言:javascript
复制
data = pd.read_csv('input.csv')

width, height = 48, 48

datapoints = data['pixels'].tolist()

#getting features for training
X = []
for xseq in datapoints:
    xx = [int(xp) for xp in xseq.split(' ')]
    xx = np.asarray(xx).reshape(width, height)
    X.append(xx.astype('float32'))

X = np.asarray(X)
X = np.expand_dims(X, -1)

#getting labels for training
y = pd.get_dummies(data['emotion']).as_matrix()

#storing them using numpy
np.save('fdataX', X)
np.save('flabels', y)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/61124739

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档