PoseEstimationForMobile
PoseEstimationForMobile copied to clipboard
how to train hourglass model??
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 `
Oh, it similar to mv2_cpm. Pls check the networks.py again.