├── .gitignore ├── README.md ├── app.py ├── applications └── balloon │ └── config.py ├── doc ├── demo_api.png ├── head.png ├── infer_flow.gif ├── 基本流程.svg ├── 数据流动.svg └── 独占资源分配器.svg ├── pyinfer ├── __init__.py ├── core │ ├── __init__.py │ ├── build.py │ ├── engine │ │ ├── __init__.py │ │ └── engine.py │ ├── hook │ │ ├── __init__.py │ │ └── hooks.py │ ├── infer │ │ ├── __init__.py │ │ ├── base.py │ │ └── detection.py │ ├── job.py │ └── mono_allocator.py └── utils │ ├── __init__.py │ ├── common │ ├── __init__.py │ ├── config.py │ ├── logger.py │ └── registry.py │ ├── detection │ ├── __init__.py │ ├── bbox.py │ └── nms.py │ └── functional │ ├── __init__.py │ ├── coco.py │ ├── slice.py │ └── traits.py ├── requirments.txt ├── static ├── README.md ├── redoc.standalone.js ├── swagger-ui-bundle.js └── swagger-ui.css ├── tools ├── generate_gif.py ├── mmdet_export_onnx │ └── balloon │ │ ├── export_yolox_onnx.py │ │ └── yolox_s_8x8_300e_coco.py ├── slice_coco.py └── visual_coco.py └── workspace ├── balloon.jpg └── infer_balloon.jpg /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/app.py -------------------------------------------------------------------------------- /applications/balloon/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/applications/balloon/config.py -------------------------------------------------------------------------------- /doc/demo_api.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/demo_api.png -------------------------------------------------------------------------------- /doc/head.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/head.png -------------------------------------------------------------------------------- /doc/infer_flow.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/infer_flow.gif -------------------------------------------------------------------------------- /doc/基本流程.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/基本流程.svg -------------------------------------------------------------------------------- /doc/数据流动.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/数据流动.svg -------------------------------------------------------------------------------- /doc/独占资源分配器.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/doc/独占资源分配器.svg -------------------------------------------------------------------------------- /pyinfer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/__init__.py -------------------------------------------------------------------------------- /pyinfer/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/__init__.py -------------------------------------------------------------------------------- /pyinfer/core/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/build.py -------------------------------------------------------------------------------- /pyinfer/core/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/engine/__init__.py -------------------------------------------------------------------------------- /pyinfer/core/engine/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/engine/engine.py -------------------------------------------------------------------------------- /pyinfer/core/hook/__init__.py: -------------------------------------------------------------------------------- 1 | from .hooks import DrawBBoxHook 2 | -------------------------------------------------------------------------------- /pyinfer/core/hook/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/hook/hooks.py -------------------------------------------------------------------------------- /pyinfer/core/infer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/infer/__init__.py -------------------------------------------------------------------------------- /pyinfer/core/infer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/infer/base.py -------------------------------------------------------------------------------- /pyinfer/core/infer/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/infer/detection.py -------------------------------------------------------------------------------- /pyinfer/core/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/job.py -------------------------------------------------------------------------------- /pyinfer/core/mono_allocator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/core/mono_allocator.py -------------------------------------------------------------------------------- /pyinfer/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyinfer/utils/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyinfer/utils/common/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/common/config.py -------------------------------------------------------------------------------- /pyinfer/utils/common/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/common/logger.py -------------------------------------------------------------------------------- /pyinfer/utils/common/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/common/registry.py -------------------------------------------------------------------------------- /pyinfer/utils/detection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyinfer/utils/detection/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/detection/bbox.py -------------------------------------------------------------------------------- /pyinfer/utils/detection/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/detection/nms.py -------------------------------------------------------------------------------- /pyinfer/utils/functional/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyinfer/utils/functional/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/functional/coco.py -------------------------------------------------------------------------------- /pyinfer/utils/functional/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/functional/slice.py -------------------------------------------------------------------------------- /pyinfer/utils/functional/traits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/pyinfer/utils/functional/traits.py -------------------------------------------------------------------------------- /requirments.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/requirments.txt -------------------------------------------------------------------------------- /static/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/static/README.md -------------------------------------------------------------------------------- /static/redoc.standalone.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/static/redoc.standalone.js -------------------------------------------------------------------------------- /static/swagger-ui-bundle.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/static/swagger-ui-bundle.js -------------------------------------------------------------------------------- /static/swagger-ui.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/static/swagger-ui.css -------------------------------------------------------------------------------- /tools/generate_gif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/tools/generate_gif.py -------------------------------------------------------------------------------- /tools/mmdet_export_onnx/balloon/export_yolox_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/tools/mmdet_export_onnx/balloon/export_yolox_onnx.py -------------------------------------------------------------------------------- /tools/mmdet_export_onnx/balloon/yolox_s_8x8_300e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/tools/mmdet_export_onnx/balloon/yolox_s_8x8_300e_coco.py -------------------------------------------------------------------------------- /tools/slice_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/tools/slice_coco.py -------------------------------------------------------------------------------- /tools/visual_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/tools/visual_coco.py -------------------------------------------------------------------------------- /workspace/balloon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/workspace/balloon.jpg -------------------------------------------------------------------------------- /workspace/infer_balloon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzyon/pyInfer/HEAD/workspace/infer_balloon.jpg --------------------------------------------------------------------------------