2

我已经为我自己的数据集尝试了这个模型(“ https://github.com/LZQthePlane/Online-Realtime-Action-Recognition-based-on-OpenPose ”)(我已经给出了坐姿动作视频并转换为csv文件) . CSV 文件包含 1000 条记录。现在在 action_enum.py 我只有 sat action 。在 train.py 文件中,我更改了

encoder_Y = [0]*744 + [1]*722 + [2]*815 + [3]*1008

encoder_Y = [0]*1000(仅用于坐下动作)。更改 train.py 文件后,出现此错误。

在此处输入图像描述

4

1 回答 1

0

由于您只有一个类,因此您应该将模型更改为仅输出一个值。

Action/training/train.py

# build keras model
model = Sequential()
model.add(Dense(units=128, activation='relu'))
model.add(BatchNormalization())
model.add(Dense(units=64, activation='relu'))
model.add(BatchNormalization())
model.add(Dense(units=16, activation='relu'))
model.add(BatchNormalization())
# change 4 to 1
model.add(Dense(units=1, activation='softmax'))
于 2019-03-11T07:26:32.020 回答