外部添加依赖文件

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

@@ -33,4 +33,3 @@ setup(
     dependency_links=dependency_links,
     author_email='Your address email (eq. you@example.com)'
 )
-