ValueError:检查目标时发生错误:预期density_4的形状为(4,),但数组的形状为(1,)

时间:2019-03-11 06:39:19

标签: python pose-estimation openpose

我已经为自己的数据集尝试了该模型(“ 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文件后,出现此错误。

enter image description here

1 个答案:

答案 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'))