Skip to content

O
onnx_model
Commit 6a09db2a authored by wangpeng's avatar wangpeng
Browse files
Options

外部添加依赖文件

parent a68b5c32
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 8 deletions
onnx_infer/main.py
View file @ 6a09db2a
...@@ -56,11 +56,9 @@ class Main: ...@@ -56,11 +56,9 @@ class Main:
return color, bbox, category return color, bbox, category
# main test # # main test
colorOnnx_path = 'color.onnx'
yoloOnnx_path = 'yolo3.onnx'
def get_result(image_arr): def get_result(colorOnnx_path,yoloOnnx_path,image_arr):
m = Main(colorOnnx_path, yoloOnnx_path, image_arr) m = Main(colorOnnx_path, yoloOnnx_path, image_arr)
color, bbox, category = m.colorAndbboxAndcategory() color, bbox, category = m.colorAndbboxAndcategory()
return color, bbox, category return color, bbox, category
......
setup.py
View file @ 6a09db2a
...@@ -21,9 +21,9 @@ setup( ...@@ -21,9 +21,9 @@ setup(
download_url='https://github.com/Your github username/mxnet_insight/tarball/' + __version__, download_url='https://github.com/Your github username/mxnet_insight/tarball/' + __version__,
license='BSD', license='BSD',
classifiers=[ classifiers=[
'Development Status :: 3 - Alpha', 'Development Status :: 3 - Alpha',
'Intended Audience :: Developers', 'Intended Audience :: Developers',
'Programming Language :: Python :: 3', 'Programming Language :: Python :: 3',
], ],
keywords='', keywords='',
packages=['onnx_infer'], packages=['onnx_infer'],
...@@ -33,4 +33,3 @@ setup( ...@@ -33,4 +33,3 @@ setup(
dependency_links=dependency_links, dependency_links=dependency_links,
author_email='Your address email (eq. you@example.com)' author_email='Your address email (eq. you@example.com)'
) )
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment