happynear / caffe-windows

Configure Caffe in one hour for Windows users.
Other
1.32k stars 650 forks source link

transpose is necessary in MTCNN? #290

Closed twmht closed 5 years ago

twmht commented 5 years ago

Hi,

In pycaffe implementation, the image is transposed before forwarding. (https://github.com/kuangliu/pycaffe-mtcnn/blob/master/main.py#L293)

But in the demo code of MTCNN you provided. I can't see any transpose operation.

Two questions,

  1. the MTCNN learned weight you provided is identical with original Matlab Caffe (https://github.com/kpzhang93/MTCNN_face_detection_alignment/tree/master/code/codes/MTCNNv1/model)?
  2. Why not transpose the image before forwarding?
happynear commented 5 years ago

I do that in MemoryData layer. https://github.com/happynear/MTCNN_face_detection_alignment/blob/master/code/codes/MTCNNv2/model/det1-memory.prototxt#L12

happynear commented 5 years ago

The model is identical with the original matlab version.

twmht commented 5 years ago

@happynear

thank you!