接上一篇教程:
1.为了方便的调用图片而不是直接去修改源码,这里讲解模块argparse的使用
- argparse模块让用户编写友好的命令行接口变得容易
- 具体使用命令:
1. import argparse
2. parser = argparse.ArgumentParser()
# 获取所有参数,就是像图片路径之类可以不用在程序本体修改,具体看例子
3. parser.add_argument('name', help='这是提示语', type=int)
# 添加参数,type可以加也可以没有
parser.add_argument('name2', help='第二个参数', type=int)
4. args = parser.parse_args()
# 解析所有参数,我把它理解为把参数集中在args中可以像调用类中属性一样调用
演示, 注意此时需要在parcharm终端运行
在左下角可以找到,当然直接命令行运行也是可以的
例:
import argparse
parser = argparse.ArgumentParser() # 获取所有参数
parser.add_argument('number1', help='the first num', type=int)
parser.add_argument('number2', help='the second num', type=int)
args = parser.parse_args() # 解析所有参数
print('the first one:', args.number1)
print('the second one', args.number2)
print('all parameters:', args)
- 然后在终端运行:
python 1.py 3 5
2. argparse 读取图片的两种方式
import cv2
import argparse
# argparse 的使用方式比较固定
parser = argparse.ArgumentParser() #获取参数
parser.add_argument('path', help='path of image') # 添加参数
args = parser.parse_args() # 解析参数
# 加载图片,方式一
img = cv2.imread(args.path)
cv2.imshow('chutian', img)
# 加载图片,方式二
arg_dict = vars(parser.parse_args()) # 在解析参数时,把参数变为字典形式
img2 = cv2.imread(arg_dict['path'])
cv2.imshow('chutian2', img2)
cv2.waitKey(0)
cv2.destroyAllWindows()
此时就可以了。我个人更喜欢字典形式的调用,因为比较直观
2. 用argparse实现图片的读取,处理, 保存
import cv2
import argparse
parser = argparse.ArgumentParser() # 获取参数
parser.add_argument('input_path')
parser.add_argument('output_path')
args = vars(parser.parse_args()) # 将解析的参数用字典形式保存
# 读取
img = cv2.imread(args['input_path'])
# 灰度处理
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 保存图片
cv2.imwrite(args['output_path'], img_gray)
- 运行:
注意前后图片不要重名
值得一提的是,这个灰度处理与原先只对图片进行灰度读取是一样的
img = cv2.imread('path', IMREAD_GRAYSCALE)
- 读取摄像头视频文件
import cv2
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('cam_index', help='摄像头的编号', type=int)
# 一般摄像头的编号从0开始,然后外设加1
args = vars(parser.parse_args())
# 打开摄像头的视频
capture = cv2.VideoCapture(args['cam_index'], cv2.CAP_DSHOW)
# 好像是微软需要加这个cv2.CAP_DSHOW ,不加会错误
# 获取帧的尺寸 , 完全是固定的用法
frame_width = capture.get(cv2.CAP_PROP_FRAME_WIDTH)
frame_height = capture.get(cv2.CAP_PROP_FRAME_HEIGHT)
# 看看视频的fps,就是每秒有多少帧
fps = capture.get(cv2.CAP_PROP_FPS)
print(fps) # 这个地方一直出错不知是为什么,始终显示0
# 因为之前已经开了摄像头了
while capture.isOpened():
ret, frame = capture.read()
# cap.read()按帧读取视频,ret,frame是获cap.read()方法的两个返回值。其中ret是布尔值
#如果读取帧是正确的则返回True,如果文件读取到结尾,它的返回值就为False。
# frame就是每一帧的图像,是个三维矩阵。
# 把捕获的帧变为灰度图
gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.imshow('frame', frame)
cv2.imshow('gray_frame', gray_frame)
if cv2.waitKey(20)==27: # esc的Ascll码
break
capture.release()
cv2.destroyAllWindows()
终端输入:
python 1.py 0
运行效果如图: