我已经为自己的数据集尝试了该模型(“ https://github.com/LZQthePlane/Online-Realtime-Action-Recognition-based-on-OpenPose”)(我已经提供了坐姿视频并转换为csv文件)。 CSV文件包含1000条记录。 现在在action_enum.py中,我只有坐姿。 在train.py文件中,我更改了
encoder_Y = [0]*744 + [1]*722 + [2]*815 + [3]*1008
到
encoder_Y = [0]*1000
(仅适用于坐姿)。
更改train.py文件后,出现此错误。
答案 0 :(得分: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'))