search

edvardHua / PoseEstimationForMobile

:dancer: Real-time single person pose estimation for Android and iOS.
Apache License 2.0
1k stars 267 forks source link

how to train hourglass model?? #16

Open lvchigo opened 6 years ago

lvchigo commented 6 years ago

From this(PoseEstimationForMobile/training/src/networks.py) file ,we just can see mv2_cpm config. So, how to train hourglass model?? ` import network_mv2_cpm

def get_network(type, input, trainable=True): if type == 'mv2_cpm': net, loss = network_mv2_cpm.build_network(input, trainable) return net, loss `

edvardHua commented 6 years ago

Oh, it similar to mv2_cpm. Pls check the networks.py again.