SSHA, SSH with Alignment
Result
How To Use
0. install mxnet and opencv for python version
1. clone SSHA
git clone https://github.com/ElegantGod/SSHA
2. make cython
cd SSHA && make
3. run it
python test_kpoint.py
Category: Python / Deep Learning |
Watchers: 10 |
Star: 144 |
Fork: 35 |
Last update: Jun 22, 2022 |
git clone https://github.com/ElegantGod/SSHA
cd SSHA && make
python test_kpoint.py
Do you release the training code of SSH with Alignment ?
Hi, congrats for the great work!
Would you mind adding a bit of documentations of how to run it, what the dependencies are and general info on how to contribute to the project?
It would help a lot of people 👍
hi,the pretrain model has very good performance on some image,it was a brilliant work. which dataset you use for the detection loss and which dataset you use for the key point loss?
env: gtx1080,cuda10
Our detection is second level,also 1face .
python test_kpoint.py [11:21:45] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.3.0. Attempting to upgrade... [11:21:45] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded! (900, 1200, 3) data.shape: (1, 3, 910, 1210) getting 1.0 32 0 9 (1, 3, 910, 1210) [11:21:48] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) [11:21:54] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) getting 1.0 16 3 9 (1, 3, 910, 1210) getting 1.0 8 6 9 (1, 3, 910, 1210) ('detection uses', 7.341606, 'seconds') ('find', 81, 'faces')
[11:27:31] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.3.0. Attempting to upgrade... [11:27:31] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded! (1160, 2000, 3) ('resize to', (936, 1608, 3)) data.shape: (1, 3, 936, 1608) getting 1.0 32 0 9 (1, 3, 936, 1608) [11:27:34] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) [11:27:41] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) getting 1.0 16 3 9 (1, 3, 936, 1608) getting 1.0 8 6 9 (1, 3, 936, 1608) ('detection uses', 8.548407, 'seconds') ('find', 13, 'faces')
[11:28:50] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.3.0. Attempting to upgrade... [11:28:50] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded! (972, 1094, 3) data.shape: (1, 3, 982, 1104) getting 1.0 32 0 9 (1, 3, 982, 1104) [11:28:54] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) [11:29:00] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) getting 1.0 16 3 9 (1, 3, 982, 1104) getting 1.0 8 6 9 (1, 3, 982, 1104) ('detection uses', 7.417773, 'seconds') ('find', 15, 'faces')
[13:47:35] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.3.0. Attempting to upgrade... [13:47:35] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded! (720, 856, 3) data.shape: (1, 3, 730, 866) getting 1.0 32 0 9 (1, 3, 730, 866) [13:47:39] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) [13:47:44] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable) getting 1.0 16 3 9 (1, 3, 730, 866) getting 1.0 8 6 9 (1, 3, 730, 866) ('detection uses', 5.2321, 'seconds') ('find', 1, 'faces')
Thanks for your sharing. l have a problem when l want to deploy this to windows system, how to complie cython file in windows system? Can you help me, Thanks.
Thanks for your sharing I want to ask why you do "cv2.copyMakeBorder" before detect in test_kpoint.py? I have no idea. Could you teach me? Thanks.
which dataset?how to train?
hi thanks the great job. I have a problem when i wanna to inference the detector. follow is the error info:
"mxnet.base.MXNetError: [16:26:09] src/operator/nn/./cudnn/cudnn_convolution-inl.h:713: Check failed: e == CUDNN_STATUS_SUCCESS (2 vs. 0) cuDNN: CUDNN_STATUS_ALLOC_FAILED
"
Do you have any idea about this?
Dear author: I want to know what dataset and landmarks annotation form of you project, pleas tell me??