# Load tensorflow model
print('--> Loading model')
ret = rknn.load_caffe(model='openpose-master/openpose-master/models//pose/coco/pose_deploy_linevec.prototxt', proto='caffe',
blobs='openpose-master/openpose-master/models//pose/coco/pose_iter_440000.caffemodel')
if ret != 0:
print('Load model failed!')
exit(ret)
print('done')
# Build model
print('--> Building model')
ret = rknn.build(do_quantization=True, dataset='./dataset.txt', pre_compile=False)
if ret != 0:
print('Build model failed!')
exit(ret)
print('done')
# Export rknn model
print('--> Export RKNN model')
ret = rknn.export_rknn('./0.rknn')
if ret != 0:
print('Export model failed!')
exit(ret)
print('done')