search

jiang-du / BlazePose-tensorflow

A third-party Tensorflow Implementation for paper "BlazePose: On-device Real-time Body Pose tracking".
Apache License 2.0
111 stars 24 forks source link

How to set train_mode for testing mode after finetuning model? #20

Open Blankit opened 3 years ago

Blankit commented 3 years ago
  1. train_mode=0 ValueError: Dimensions must be equal, but are 14 and 128 for '{{node mean_squared_error/SquaredDifference}} = SquaredDifference[T=DT_FLOAT](blaze_pose/sequential_76/reshape/Reshape, IteratorGetNext:1)' with input shapes: [?,14,3], [?,128,128,14].
  2. train_mode=1,output is [ 8.4 16.7 29.8 27.8 16.1 9.3 6.3 9.9 25.5 21.4 14.2 7. 34.7 23.7] Average = 17.914286%