I am trying to compile my customized yolo model. I have created an Inference model as well and generated .pb file. However, when I compile with movidius I get following error:
res.shape: (1, 13, 13, 3, 8) Traceback (most recent call last): File "/usr/local/bin/mvNCCompile", line 206, in create_graph(args.network, args.image, args.inputnode, args.outputnode, args.outfile, args.nshaves, args.inputsize, args.weights, args.explicit_concat, args.ma2480, args.scheduler, args.new_parser, args.cpp, args) File "/usr/local/bin/mvNCCompile", line 185, in create_graph load_ret = load_network(args, parser, myriad_config) File "/usr/local/bin/ncsdk/Controllers/Scheduler.py", line 146, in load_network parse_ret = parse_tensor(arguments, myriad_conf) File "/usr/local/bin/ncsdk/Controllers/TensorFlowParser.py", line 1751, in parse_tensor res = res.reshape(1, 1, res.shape[0]) ValueError: cannot reshape array of size 4056 into shape (1,1,1)
Do you have any suggestions on it that how to solve this?
My actions:
-> Trained custom yolo model and saved weights file.
->Restored meta file and created NN.ckpt files.
-> Used these checkpoint files to freeze the network
->Tried to compile the generated frozen graph with movidius and got the above error,
If you have any suggestions on it please let me know...
Thanks
Hi,
I am trying to compile my customized yolo model. I have created an Inference model as well and generated .pb file. However, when I compile with movidius I get following error:
res.shape: (1, 13, 13, 3, 8) Traceback (most recent call last): File "/usr/local/bin/mvNCCompile", line 206, in create_graph(args.network, args.image, args.inputnode, args.outputnode, args.outfile, args.nshaves, args.inputsize, args.weights, args.explicit_concat, args.ma2480, args.scheduler, args.new_parser, args.cpp, args) File "/usr/local/bin/mvNCCompile", line 185, in create_graph load_ret = load_network(args, parser, myriad_config) File "/usr/local/bin/ncsdk/Controllers/Scheduler.py", line 146, in load_network parse_ret = parse_tensor(arguments, myriad_conf) File "/usr/local/bin/ncsdk/Controllers/TensorFlowParser.py", line 1751, in parse_tensor res = res.reshape(1, 1, res.shape[0]) ValueError: cannot reshape array of size 4056 into shape (1,1,1)
Do you have any suggestions on it that how to solve this? My actions: -> Trained custom yolo model and saved weights file. ->Restored meta file and created NN.ckpt files. -> Used these checkpoint files to freeze the network ->Tried to compile the generated frozen graph with movidius and got the above error,
If you have any suggestions on it please let me know... Thanks