image_size is not set properly for PNet test_mode in gen_hard_example.py

AITTSMD / MTCNN-Tensorflow

Reproduce MTCNN using Tensorflow

1.51k stars 713 forks source link

image_size is not set properly for PNet test_mode in gen_hard_example.py #160

Closed sumsuddin closed 6 years ago

sumsuddin commented 6 years ago

In this script When I pass PNet as test_mode like this,

$ python gen_hard_example.py --test_mode PNet

The image_size variable is not set to 24. No matter what param I pass as test_mode the image_size is always 48. So it is not generating input images for training RNet.

Please have a look or correct me if I am wrong somewhere. TIA

liuyunwww commented 6 years ago

https://github.com/AITTSMD/MTCNN-Tensorflow/blob/master/prepare_data/gen_hard_example.py Line 215 Change ONet to RNet

sumsuddin commented 6 years ago

Thanks, @liuyunwww !! Yes, I did the same to make it work. Here is a pull request where I tried to fix the issue and set the image size accordingly to the flag. Can you have a look and suggest me to improve?