一、输出图片保存
源代码里没有图片保存,需要在demo.py自己添加一句代码保存输出的图片:
添加代码如129行所示:cv2.imwrite('要保存的文件名称',img)
二、行人框格式调整
想要修改行人框的颜色和线条粗细,在demo.py
124行cv2.rectangle函数中修改参数,(0,225,0)修改成自己想要的颜色对应的RGB数值,线条粗细修改thickness值即可。
三、骨架图格式调整
在pose.py中62行修改cv2.line中最后一个参数,改变线条粗细。
颜色修改更简单,pose.py中19行把color改为自己想要的颜色RGB值即可。最好选颜色鲜艳一点的,不然有点看不清!
好了,关于图片的就这么多!保存视频的我再研究一下!