xingyizhou / DeepModel

Code repository for Model-based Deep Hand Pose Estimation
GNU General Public License v3.0
111 stars 43 forks source link

accuracy layer on training phase #19

Open weiguochow opened 6 years ago

weiguochow commented 6 years ago

Hi, xingyi Than you for your great open source code. I have reproduced the results as you show on the repository. Furthermore, I want to add accuracy layer in DeepModel.prototxt file .

layer {
            name: "accuracy"
            type: "Accuracy"
            bottom: "DeepHandModelxyz"
            bottom: "joint"
            top: "accuracy"
            include {
                  phase: "TEST"
                   }
          }

However, there is error say that the dimension of jonitand DeepHandModelxyz doesn't match.

Number of labels must match number of predictions; e.g., if label axis == 1 and prediction shape is (N, C, H, W), label count (number of labels) must be N*H*W, with integer values in {0, 1, ..., C-1}

Is it maybe the sake of hd5 format data format? Can you give me some help? I am looking forward to your replay. Thank you!!!