It looks like you're new here. If you want to get involved, click one of these buttons!Sign In
It looks like you're new here. If you want to get involved, click one of these buttons!
I'm a newer to use ncs. About one month ago I purchased my ncs stick but with no idea to apply my own model.
Goal and What I did
Goal: I'm trying to train a vgg_16 model to detect object, I'll be very happy if faster-rcnn is also available.
What I did: I have a stable version of vgg16 model trained by Faster RCNN with my own dataset. I generated a vgg16 graphDef by using export_inference_graph.py, and then use freeze_graph.py but something seems wrong (because I met rhs lhs issue) and I even don't know what is the output node for vgg16. Then I thought maybe it's important to train a smaller model by using TensorFlow's Object Detection API firstly, and that is my first step.
I'd like to apply Tensorflow's Object detection API with ssd_mobilenet v1 onto ncs. By following https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_locally.md I got a saved_model.pb file from ssd_mobile_v1.ckpt, then I ran
mvNCCompile saved_model.pb -in=input -on=MobilenetV1/Predictions/Reshape_1 -is 224 224 -o graph
Maybe there are some error arguments, I got an issue like
Traceback (most recent call last): File "/usr/local/bin/mvNCCompile", line 118, in <module> create_graph(args.network, args.inputnode, args.outputnode, args.outfile, args.nshaves, args.inputsize, args.weights) File "/usr/local/bin/mvNCCompile", line 104, in create_graph net = parse_tensor(args, myriad_config) File "/usr/local/bin/ncsdk/Controllers/TensorFlowParser.py", line 210, in parse_tensor graph_def.ParseFromString(f.read()) google.protobuf.message.DecodeError: Error parsing message
I got this unexpected error in my first step. Could anyone give me some direction? Thanks.