search

junxnone / tio

Log
Other
10 stars 5 forks source link

Realtime Multi-Person Pose Estimation - Tensorflow #330

Open junxnone opened 5 years ago

junxnone commented 5 years ago

296

基于TensorFlow的openpose 实现

Reference

SetupEnv

virtualenv optfenv
source optfenv/bin/activate
pip install tensorpack tqdm tensorflow-gpu pandas requests opencv-python matplotlib

Install cocoapi #257

Download vgg19 weights & COCO2017

mkdir -p checkpoints/vgg/
cd checkpoints/vgg/
wget http://download.tensorflow.org/models/vgg_19_2016_08_28.tar.gz
tar zvxf vgg_19_2016_08_28.tar.gz

Training

python train.py \
--img_path COCO/images/ \
--annot_path COCO/annotations/ \
--checkpoint_path checkpoints/vgg/
junxnone commented 5 years ago

跑到13% 挂住了？

 12%|███████████████████████████▎
                                       | 1399/11828 [21:58<1:48:22,  1.60it/s][2019-01-30 10:39:29,716] [train] [INFO] echos=0.118363, setp=1400, total_loss=405.849335, lr=0.000100
 13%|█████████████████████████████▎
                                       | 1499/11828 [23:06<1:47:54,  1.60it/s][2019-01-30 10:40:37,063] [train] [INFO] echos=0.126818, setp=1500, total_loss=308.516846, lr=0.000100
 13%|██████████████████████████████▉
                                       | 1583/11828 [24:00<1:48:34,  1.57it/s]