dmlc / dmlc/web-data

tensorflow pre-trained object detection model supported?

Open
#146 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
84
Forks
151
PR merge metrics
No merged PRs in 30d

Description

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'])

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.