You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
hi, guys,
any guys can point out how to use tensorflow pre-trained object detection model?
I checked the doc, it seems just TVM support classification for tensorflow framework.
now I want to use pre-trained model for tensorflow (for example faster_rcnn_resnet101_coco_2018_01_28.tar.gz or ssd_mobilenet_v1_coco_11_06_2017.tar.bz2), and used script "tvm/tutorials/nnvm/from_tensorflow.py" with some modification (just point to these pre-trained models).
but I get some errors:
Traceback (most recent call last):
File "from_tensorflow.py", line 98, in
graph_def = nnvm.testing.tf.AddShapesToGraphDef(sess, 'softmax')
File "/root/ai/tvm/nnvm/python/nnvm/testing/tf.py", line 70, in AddShapesToGraphDef
[out_node],
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 227, in convert_variables_to_constants
inference_graph = extract_sub_graph(input_graph_def, output_node_names)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 171, in extract_sub_graph
_assert_nodes_are_present(name_to_node, dest_nodes)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 131, in _assert_nodes_are_present
assert d in name_to_node, "%s is not in graph" % d
AssertionError: softmax is not in graph
if I comment out following codes,
" with tf.Session() as sess:
graph_def = nnvm.testing.tf.AddShapesToGraphDef(sess, 'softmax')"
and get below errors:
Traceback (most recent call last):
File "from_tensorflow.py", line 123, in
sym, params = nnvm.frontend.from_tensorflow(graph_def)
File "/root/ai/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1503, in from_tensorflow
sym, params = g.from_tensorflow(graph, layout, shape, outputs)
File "/root/ai/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1163, in from_tensorflow
"The following operators are not implemented: {}".format(missing_operators))
NotImplementedError: The following operators are not implemented: set([u'Slice', u'TopKV2', u'Sqrt', u'CropAndResize', u'Exit', u'Tile', u'TensorArrayGatherV3', u'Max', u'NonMaxSuppressionV2', u'LogicalAnd', u'Assert', u'TensorArraySizeV3', u'TensorArrayWriteV3', u'TensorArrayReadV3', u'All', u'LoopCond', u'Merge', u'Switch', u'Exp', u'Enter', u'Where', u'Round', u'NextIteration', u'TensorArrayV3', u'TensorArrayScatterV3', u'ZerosLike', u'Select', u'Size'])
The text was updated successfully, but these errors were encountered:
hi, guys,
any guys can point out how to use tensorflow pre-trained object detection model?
I checked the doc, it seems just TVM support classification for tensorflow framework.
now I want to use pre-trained model for tensorflow (for example faster_rcnn_resnet101_coco_2018_01_28.tar.gz or ssd_mobilenet_v1_coco_11_06_2017.tar.bz2), and used script "tvm/tutorials/nnvm/from_tensorflow.py" with some modification (just point to these pre-trained models).
but I get some errors:
Traceback (most recent call last):
File "from_tensorflow.py", line 98, in
graph_def = nnvm.testing.tf.AddShapesToGraphDef(sess, 'softmax')
File "/root/ai/tvm/nnvm/python/nnvm/testing/tf.py", line 70, in AddShapesToGraphDef
[out_node],
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 227, in convert_variables_to_constants
inference_graph = extract_sub_graph(input_graph_def, output_node_names)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 171, in extract_sub_graph
_assert_nodes_are_present(name_to_node, dest_nodes)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/graph_util_impl.py", line 131, in _assert_nodes_are_present
assert d in name_to_node, "%s is not in graph" % d
AssertionError: softmax is not in graph
if I comment out following codes,
" with tf.Session() as sess:
graph_def = nnvm.testing.tf.AddShapesToGraphDef(sess, 'softmax')"
and get below errors:
Traceback (most recent call last):
File "from_tensorflow.py", line 123, in
sym, params = nnvm.frontend.from_tensorflow(graph_def)
File "/root/ai/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1503, in from_tensorflow
sym, params = g.from_tensorflow(graph, layout, shape, outputs)
File "/root/ai/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1163, in from_tensorflow
"The following operators are not implemented: {}".format(missing_operators))
NotImplementedError: The following operators are not implemented: set([u'Slice', u'TopKV2', u'Sqrt', u'CropAndResize', u'Exit', u'Tile', u'TensorArrayGatherV3', u'Max', u'NonMaxSuppressionV2', u'LogicalAnd', u'Assert', u'TensorArraySizeV3', u'TensorArrayWriteV3', u'TensorArrayReadV3', u'All', u'LoopCond', u'Merge', u'Switch', u'Exp', u'Enter', u'Where', u'Round', u'NextIteration', u'TensorArrayV3', u'TensorArrayScatterV3', u'ZerosLike', u'Select', u'Size'])
The text was updated successfully, but these errors were encountered: