search

last-one / Pytorch_Realtime_Multi-Person_Pose_Estimation

Pytorch version of Realtime Multi-Person Pose Estimation project
MIT License
216 stars 70 forks source link

Faster augmentation #1

Open anatolix opened 6 years ago

anatolix commented 6 years ago

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.

last-one commented 6 years ago

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