1 Star 0 Fork 48

Yonja / hand-keras-yolo3-recognize

加入 Gitee
与超过 600 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
SaveImg_graphviz.py 1.39 KB
一键复制 编辑 Web IDE 原始数据 按行查看 历史
cungudafa 提交于 2020-08-08 09:50 . sign
#!/usr/bin/env python
import time
from pycallgraph import PyCallGraph
from pycallgraph.output import GraphvizOutput
from pycallgraph import Config
from pycallgraph import GlobbingFilter
from pose_hand import getImgInfo #引入外部函数和类
from pose.coco import general_coco_model
from yolo import YOLO
#也可以直接写在这里def getData():...
def main():
config = Config()
# 关系图中包括(include)哪些函数名。
#如果是某一类的函数,例如类gobang,则可以直接写'gobang.*',表示以gobang.开头的所有函数。(利用正则表达式)。
config.trace_filter = GlobbingFilter(include=[
'main',
'pycallgraph.*',
'*.secret_function',
])
graphviz = GraphvizOutput()
graphviz.output_file = 'basic.png'#图片名称
with PyCallGraph(output=graphviz):
# coco
modelpath = "model/"
start = time.time()
pose_model = general_coco_model(modelpath) # 1.加载模型
print("[INFO]Pose Model loads time: ", time.time() - start)
# yolo
start = time.time()
_yolo = YOLO() # 1.加载模型
print("[INFO]yolo Model loads time: ", time.time() - start)
img_path = 'D:/myworkspace/dataset/My_test/img/a_img/airplane_30.jpg'
getImgInfo(img_path,pose_model,_yolo)
if __name__ == '__main__':
main()

评论 ( 0 )

你可以在登录后,发表评论

Python
1
https://gitee.com/yonja/hand-keras-yolo3-recognize.git
git@gitee.com:yonja/hand-keras-yolo3-recognize.git
yonja
hand-keras-yolo3-recognize
hand-keras-yolo3-recognize
master

搜索帮助

105716 1d94204e 1850385 105716 2d26be5c 1850385