model_fusion_tool:模型融合工具

时间:2024-03-27 10:36:55
【文件属性】:

文件名称:model_fusion_tool:模型融合工具

文件大小:2.47MB

文件格式:ZIP

更新时间:2024-03-27 10:36:55

Python

model_fusion_tool 一种基于协议缓冲区的模型融合工具,可以将多个模型融合到一起,以达到减少多次加载消耗的资源和时间,同时节约推理时间。 适用场景 适用于有级联关系的不同模型间的融合操作,如目标检测后提取图案内容后,再进行其他任务的操作。 适用于有相同输入的,不同分支的任务。如检测,识别等一起做。(单种任务模型已有,可融合后适用于复杂任务,避免重新设计,训练模型) 其他(你想怎么连接就怎么来) 支持的类型 图:pb定义的图,可修改结点前缀,删除,修改部分结点 数据:数据结点,可预期函数 风俗:自定义结点 示例 examples目录下 examples/example_custom.pbtxt定义了一个自定义结点。 python generate.py examples/example_custom.pbtxt examples/example_custom.txt 或者 p


【文件预览】:
model_fusion_tool-main
----functions()
--------__pycache__()
--------make_anchor.py(2KB)
----bin()
--------protoc(11.74MB)
----compile.sh(65B)
----call_node.py(481B)
----__pycache__()
--------call_node.cpython-36.pyc(926B)
----include()
--------google()
----generate.py(695B)
----.DS_Store(8KB)
----nodes()
--------__pycache__()
--------__init__.py(182B)
--------data_node.py(1KB)
--------graph_node.py(2KB)
--------custom_node.py(4KB)
--------base_node.py(316B)
----examples()
--------example_custom.pbtxt(308B)
--------example_data.pbtxt(196B)
--------example_data.pb(66KB)
--------example_custom.pb(185B)
--------example_graph.pbtxt(274B)
--------example_data.txt(173KB)
--------example_custom.txt(810B)
----protos()
--------__pycache__()
--------ops.proto(1KB)
--------ops_pb2.py(27KB)
----README.md(1KB)
----utils()
--------__pycache__()
--------node_util.py(6KB)
--------model_util.py(931B)

网友评论