search

jiang-du / BlazePose-tensorflow

A third-party Tensorflow Implementation for paper "BlazePose: On-device Real-time Body Pose tracking".
Apache License 2.0
113 stars 24 forks source link

最终这个模型的计算量和参数量是多大啊 #6

Open happysheep224 opened 3 years ago

happysheep224 commented 3 years ago

如题

HongChow commented 3 years ago

如题

能跑起来吗? 预训练系数在哪里?

jiang-du commented 3 years ago

You can run model.summary() to view the details.

Model: "blaze_pose"
_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
conv2d (Conv2D)              multiple                  672
_________________________________________________________________
sequential (Sequential)      (None, 128, 128, 24)      840
_________________________________________________________________
sequential_1 (Sequential)    (None, 128, 128, 24)      840
_________________________________________________________________
blaze_block (BlazeBlock)     multiple                  9936
_________________________________________________________________
blaze_block_1 (BlazeBlock)   multiple                  46272
_________________________________________________________________
blaze_block_2 (BlazeBlock)   multiple                  214464
_________________________________________________________________
blaze_block_3 (BlazeBlock)   multiple                  574176
_________________________________________________________________
sequential_28 (Sequential)   (None, 16, 16, 48)        16752
_________________________________________________________________
sequential_29 (Sequential)   (None, 16, 16, 48)        11184
_________________________________________________________________
up_sampling2d_1 (UpSampling2 multiple                  0
_________________________________________________________________
sequential_30 (Sequential)   (None, 32, 32, 48)        5616
_________________________________________________________________
up_sampling2d_2 (UpSampling2 multiple                  0
_________________________________________________________________
sequential_31 (Sequential)   (None, 64, 64, 48)        2832
_________________________________________________________________
sequential_32 (Sequential)   (None, 128, 128, 8)       872
_________________________________________________________________
sequential_33 (Sequential)   (None, 128, 128, 8)       440
_________________________________________________________________
sequential_34 (Sequential)   (None, 128, 128, 14)      1174
_________________________________________________________________
blaze_block_4 (BlazeBlock)   multiple                  46272
_________________________________________________________________
sequential_41 (Sequential)   (None, 32, 32, 96)        10272
_________________________________________________________________
blaze_block_5 (BlazeBlock)   multiple                  214464
_________________________________________________________________
sequential_49 (Sequential)   (None, 16, 16, 192)       38976
_________________________________________________________________
blaze_block_6 (BlazeBlock)   multiple                  574176
_________________________________________________________________
sequential_58 (Sequential)   (None, 8, 8, 288)         86112
_________________________________________________________________
sequential_75 (Sequential)   (None, 2, 2, 288)         1377792
_________________________________________________________________
sequential_76 (Sequential)   (None, 14, 3)             12138
=================================================================
Total params: 3,246,272
weijie1111 commented 3 years ago

checkpoint_path 有问题怎么办啊