Pytorch_Realtime_Multi-Person_Pose_Estimation icon indicating copy to clipboard operation
Pytorch_Realtime_Multi-Person_Pose_Estimation copied to clipboard

Faster augmentation

Open anatolix opened this issue 6 years ago • 1 comments

Hi.

I've made very fast augmentation algorithms for keras version of project https://github.com/anatolix/keras_Realtime_Multi-Person_Pose_Estimation It augments 140 images per second in python (C++ augmentation augments 30)

Trick is in made all augmentation(crop/scale/resize/shift) with one warpAffine and vectorized generation of heatmaps and pafs. Please take a look, if you like it i could adapt it and make pull request.

anatolix avatar Nov 16 '17 11:11 anatolix

Thank you very much. It's a wonderful work!

last-one avatar Nov 16 '17 12:11 last-one