外部添加依赖文件

onnx_infer/main.py

@@ -56,11 +56,9 @@ class Main:
         return color, bbox, category
 
 
-# main test
-colorOnnx_path = 'color.onnx'
-yoloOnnx_path = 'yolo3.onnx'
+# # main test
 
-def get_result(image_arr):
+def get_result(colorOnnx_path,yoloOnnx_path,image_arr):
     m = Main(colorOnnx_path, yoloOnnx_path, image_arr)
     color, bbox, category = m.colorAndbboxAndcategory()
     return color, bbox, category

setup.py

@@ -21,9 +21,9 @@ setup(
     download_url='https://github.com/Your github username/mxnet_insight/tarball/' + __version__,
     license='BSD',
     classifiers=[
-      'Development Status :: 3 - Alpha',
-      'Intended Audience :: Developers',
-      'Programming Language :: Python :: 3',
+        'Development Status :: 3 - Alpha',
+        'Intended Audience :: Developers',
+        'Programming Language :: Python :: 3',
     ],
     keywords='',
     packages=['onnx_infer'],
@@ -33,4 +33,3 @@ setup(
     dependency_links=dependency_links,
     author_email='Your address email (eq. you@example.com)'
 )
-