opencv-python函数
opencv-python读取、展示和存储图像
1、imshow函数
imshow函数作用是在窗口中显示图像,窗口自动适合于图像大小,我们也可以通过imutils模块调整显示图像的窗口的大小。函数官方定义如下:
cv2.imshow(windows_name, image)
函数参数一: 窗口名称(字符串)
函数参数二: 图像对象,类型是numpy中的ndarray类型,注:这里可以通过imutils模块改变图像显示大小,下面示例展示
上示例:
cv2.imshow('origin image', rgb_img) #显示原图
cv2.imshow('origin image', imutils.resize(rgb_img, 800)) #利用imutils模块调整显示图像大小
cv2.imshow('gray image', imutils.resize(gray_img, 800))
if cv2.waitKey(0) == 27:
cv2.destroyAllWindows()
2、imwrite函数
imwrite函数检图像保存到本地,官方定义:
cv2.imwrite(image_filename, image)
函数参数一: 保存的图像名称(字符串)
函数参数二: 图像对象,类型是numpy中的ndarray类型
cv2.imwrite('rgb_img.jpg', rgb_img) #将图像保存成jpg文件
cv2.imwrite('gray_img.png', gray_img) #将图像保存成png文件
3、窗口销毁函数
当我们使用imshow函数展示图像时,最后需要在程序中对图像展示窗口进行销毁,否则程序将无法正常终止,常用的销毁窗口的函数有下面两个:
(1)、cv2.destroyWindow(windows_name) #销毁单个特定窗口
参数: 将要销毁的窗口的名字
(2)、cv2.destroyAllWindows() #销毁全部窗口,无参数
那我们合适销毁窗口,肯定不能图片窗口一出现我们就将窗口销毁,这样便没法观看窗口,试想有两种方式:
(1) 让窗口停留一段时间然后自动销毁;
(2) 接收指定的命令,如接收指定的键盘敲击然后结束我们想要结束的窗口
以上两种情况都将使用cv2.waitKey函数, 首先产看函数定义:
cv2.waitKey(time_of_milliseconds)
唯一参数 time_of_milliseconds是整数,可正可负也可是零,含义和操作也不同,分别对应上面说的两种情况
(1) time_of_milliseconds > 0 :此时time_of_milliseconds表示时间,单位是毫秒,含义表示等待 time_of_milliseconds毫秒后图像将自动销毁,看以下示例
#表示等待10秒后,将销毁所有图像
if cv2.waitKey(10000):
cv2.destroyAllWindows()
#表示等待10秒,将销毁窗口名称为'origin image'的图像窗口
if cv2.waitKey(10000):
cv2.destroyWindow('origin image')
(2) time_of_milliseconds <= 0 : 此时图像窗口将等待一个键盘敲击,接收到指定的键盘敲击便会进行窗口销毁。我们可以自定义等待敲击的键盘,通过下面的例子进行更好的解释
设置waitKey(0),则表示程序会无限制的等待用户的按键事件
#当指定waitKey(0) == 27时当敲击键盘 Esc 时便销毁所有窗口
if cv2.waitKey(0) == 27:
cv2.destroyAllWindows()
#当接收到键盘敲击A时,便销毁名称为'origin image'的图像窗口
if cv2.waitKey(-1) == ord('A'):
cv2.destroyWindow('origin image')
下面给出一个完整读取、展示、保存和图像销毁示例:
import cv2
import imutils
import numpy as np
rgb_img = cv2.imread('E:/peking_rw/ocr_project/base_prehandle/img/cartoon.jpg')
gray_img = cv2.cvtColor(rgb_img, cv2.COLOR_BGR2GRAY)
cv2.imshow('origin image', imutils.resize(rgb_img, 800))
cv2.imshow('gray image', imutils.resize(gray_img, 800))
cv2.imwrite('rgb_img.jpg', rgb_img)
cv2.imwrite('gray_img.png', gray_img)
#等待一定时间自动销毁图像窗口
#if cv2.waitKey(10000):
# cv2.destroyAllWindows()
#if cv2.waitKey(10000):
# cv2.destroyWindow('origin image')
#接收特定键盘销毁图像窗口
#if cv2.waitKey(-1) == ord('A'):
# cv2.destroyWindow('origin image')
if cv2.waitKey(0) == 27:
cv2.destroyAllWindows()
给图片添加文字
ImageFont:用来加载准备阶段中下载的字体库文件
ImageDraw: 基于image对象,创建一个可以在Image实例上画线条、贴文字的对象。
import cv2
from PIL import ImageFont, ImageDraw, Image
import numpy as np
bk_img = cv2.imread("background.jpg")
#设置需要显示的字体
fontpath = "font/simsun.ttc"
font = ImageFont.truetype(fontpath, 32) #32为字体大小
img_pil = Image.fromarray(bk_img)
draw = ImageDraw.Draw(img_pil)
#绘制文字信息
draw.text((100, 300), "Hello World", font = font, fill = (255, 255, 255))
draw.text((100, 350), "你好", font = font, fill = (255, 255, 255))
bk_img = np.array(img_pil)
cv2.imshow("add_text",bk_img)
cv2.waitKey()、
#保存图片
cv2.imwrite("add_text.jpg",bk_img)
pillow的简单使用——南昌理工学院实践项目
import time
import random
import threading
import copy
import cv2 # 导入 opencv -python 库
from PIL import Image,ImageFont,ImageDraw
import numpy as np
img1 = cv2.imread('d:\\lib.jpg')
img2 = cv2.imread('d:\\lib1.png')
img2 = cv2.resize(img2, (100, 100))
img=Image.fromarray(img1)
draw = ImageDraw.Draw(img)
fontText = ImageFont.truetype('C:/Windows/Fonts/msyh.ttc', 30, encoding="utf-8")
draw.text((500, 30), '....南理文字RPG0.2的世界开始了.....', textColor=(255,0,0), font=fontText)
draw.text((535, 100), 'Hi,欢迎加入南理文字RPG0.2 ', textColor=(255,0,0), font=fontText)
img1=np.array(img)
SCRIPT_NPC_SCHOOL_SISTER = ['你好!',
'你好!',
'你是新生嘛?',
'是的',
'想要我教你魔法吗?',
'\n1、好的 \
\n2、不用了吧,我不和学姐学魔法!\n'
]
MAGIC_BOOKS = [
['风巽千叶翔龙', 1],
['泽兑鬼尘珠', 2],
['凤翔九天', 3]
]
class Body: # 人物类
LEVEL=1 #等级
EXP=0 #经验
NAME = '' # 名字
HP = 0 # 血量
MAX_HP = 0 # 最大血量
MP = 0 # 魔法值
MAX_MP = 0 # 最大魔法值
ATTACK = 1
MAGIC_ATTACK = 1.3
DEFENSE=5
HAS_MAGIC = [] # 会的魔法
EMOTION=0.5 #范围0-1,0为绝望,1为神采奕奕,(0.5为平和,此时为最佳状态)
POS_NAME = '南昌理工学院南大门' # 地点
def __init__(self, IS_HUMAN_= False,NAME_='Ice NPC',LEVEL_=1,EXP_=0, HP_=3, MP_=3, MAX_HP_=3, MAX_MP_=3, ATTACK_=1, MAGIC_ATTACK_=1.3,EMOTION_=0.5, HAS_MAGIC_=[],
POS_NAME_='南昌理工学院南大门', DEFENSE_=5,bounce_=0):
# 初始化 默认值
self.IS_HUMAN = IS_HUMAN_ # 判断是否位人类
self.LEVEL=LEVEL_
self.EXP=EXP_
self.NAME = NAME_
self.HP = HP_
self.MP = MP_
self.MAX_HP = MAX_HP_
self.MAX_MP = MAX_MP_
self.ATTACK = ATTACK_
self.MAGIC_ATTACK = MAGIC_ATTACK_
self.DEFENSE=DEFENSE_
self.EMOTION=EMOTION_
self.POS_NAME = POS_NAME_
self.HAS_MAGIC = HAS_MAGIC_
self.bounce=bounce_
def display_status(self): # 显示人物状态
print('\n\r', ' 状态:' + self.NAME, end='')
print('\n血量:' + '❤' * int(self.HP), '\n 魔法值:' + '⚓' * int(self.MP))
print('地点:' + self.POS_NAME)
if self.HAS_MAGIC == []:
print('无魔法')
else:
self.display_magic()
def display_status_fighting(self):
print('\n\r', ' 状态:' + self.NAME, end='')
print('\n血量:' + '❤' * int(self.HP), ' 魔法值:' + '⚓' * int(self.MP))
def display_magic(self): # 显示魔法
print(self.NAME + '会如下几种魔法:')
print('魔法名 ,消耗点数')
i = 0
for iMagic in self.HAS_MAGIC:
print('第', i, '个魔法:', MAGIC_BOOKS[iMagic])
i = i + 1
def talk_to(self, NPC_, NPC_SCRIPT_, delay=0.3):
# 按剧本和NPC聊天
print('在' + self.POS_NAME + '遇到 ' + NPC_.NAME)
for i in range(len(NPC_SCRIPT_)):
if i % 2 == 1: # 判断当前说话的人
TURN = self.NAME
else:
TURN = NPC_.NAME
print(TURN + ':' + NPC_SCRIPT_[i])
time.sleep(delay)
def learn_magic_from(self, NPC_):
# 向NPC学习魔法
NPC_.display_magic()
CHOOSE = input('是否向' + NPC_.NAME + '学魔法?(输入:1/2/3/no)')
if CHOOSE == '1':
print('恭喜' + self.NAME + '学会' + MAGIC_BOOKS[NPC_.HAS_MAGIC[0]][0] +
',每次使用消耗' + str(MAGIC_BOOKS[NPC_.HAS_MAGIC[0]][1]) + 'MP.')
self.HAS_MAGIC = self.HAS_MAGIC + [NPC_.HAS_MAGIC[0]]
self.display_magic()
elif CHOOSE == '2':
print('恭喜' + self.NAME + '学会' + MAGIC_BOOKS[NPC_.HAS_MAGIC[1]][0] +
',每次使用消耗' + str(MAGIC_BOOKS[NPC_.HAS_MAGIC[1]][1]) + 'MP.')
self.HAS_MAGIC = self.HAS_MAGIC + [NPC_.HAS_MAGIC[1]]
self.display_magic()
elif CHOOSE == '3':
print('恭喜' + self.NAME + '学会' + MAGIC_BOOKS[NPC_.HAS_MAGIC[2]][0] +
',每次使用消耗' + str(MAGIC_BOOKS[NPC_.HAS_MAGIC[2]][1]) + 'MP.')
self.HAS_MAGIC = self.HAS_MAGIC + [NPC_.HAS_MAGIC[2]]
self.display_magic()
elif CHOOSE == 'no':
print('由于你的固执,没有学习新的魔法.')
else:
print('不明白你说什么.')
def DEFENSE_(self, NPC):
hit = self.ATTACK * (100 - self.DEFENSE) % 100
return hit
def attack_one_time(self, NPC_, iChoose_magic_): # 攻击一下NPC
if iChoose_magic_ == -2: # 休息,且调出自己的全面状态
self.display_status()
elif iChoose_magic_ == -1: # 普通攻击
NPC_.HP = NPC_.HP - self.ATTACK
self.HP = NPC_.HP + self.DEFENSE
self.EXP = self.EXP + MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]
print('【', self.NAME + '击剑攻击' + NPC_.NAME, '造成了' + str(self.ATTACK), '点攻击】')
print('***' + self.NAME + '增加了' + str(MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]) + '点经验值***')
print('【', self.NAME + '消耗了' + str(MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]) + '点魔法】')
print('***' + self.NAME + '的经验值为' + str(self.EXP) + '***')
elif iChoose_magic_ in self.HAS_MAGIC:
# print(iChoose_magic_)
hit = self.MAGIC_ATTACK * MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]
# print('hit:',hit,',',self.MAGIC_ATTACK,',',self.HAS_MAGIC[iChoose_magic_])
NPC_.HP = NPC_.HP - hit # 魔法伤害=魔法消耗*魔法攻击力
NPC_.MP = NPC_.MP - self.HAS_MAGIC[iChoose_magic_]
NPC_.MP = NPC_.MP + self.HAS_MAGIC[iChoose_magic_] # 实现魔法无消耗
self.MP = self.MP - MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]
self.EXP = self.EXP + MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]
print('【', self.NAME + '用魔法' + MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][0] + ',攻击了' + NPC_.NAME + ',',
'造成了' + str(hit) + '点攻击!】')
print('***' + self.NAME + '增加了' + str(MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]) + '点经验值***')
print('【', self.NAME + '消耗了' + str(MAGIC_BOOKS[self.HAS_MAGIC[iChoose_magic_]][1]) + '点魔法】')
print('***' + self.NAME + '的经验值为' + str(self.EXP) + '***')
if self.EXP >= 2: # 如果经验值大于等于2 玩家即升一级
print('***' + self.NAME + '已升级') # 显示玩家已升级
self.LEVEL = self.LEVEL + 1
self.EXP = 0 # 升级后 经验清零
print('***' + self.NAME + '的等级为' + str(self.LEVEL) + '***')
# 玩家各属性提升
self.HP = self.HP + 1
self.MP = self.MP + 1
self.MAX_HP = self.MAX_HP + 1
self.MAX_MP = self.MAX_MP + 1
self.DEFENSE = self.DEFENSE + 5
self.ATTACK = self.ATTACK + 1
self.MAGIC_ATTACK = self.MAGIC_ATTACK + 0.5
# 输出玩家升级后属性值
print('***' + self.NAME + '的HP为' + str(self.HP) + '***')
print('***' + self.NAME + '的MP为' + str(self.MP) + '***')
print('***' + self.NAME + '的最大生命为' + str(self.MAX_HP) + '***')
print('***' + self.NAME + '的最大魔法为' + str(self.MAX_MP) + '***')
print('***' + self.NAME + '的普通攻击力为' + str(self.ATTACK) + '***')
print('***' + self.NAME + '的魔法攻击为' + str(self.MAGIC_ATTACK) + '***')
print('***' + self.NAME + '的防御为' + str(self.DEFENSE) + '***')
if NPC_.HP <= 0:
return 1
else:
return 0
def fight_with(self, NPC_):
# 挑战NPC
print('【', self.NAME + '开始和' + NPC_.NAME + '战斗! !', '】')
MAX_ROUND = 100
IS_DEAD = 0
for ROUND in range(MAX_ROUND):
if (ROUND % 2 == 0):
self.display_status_fighting()
self.display_magic()
NPC_.display_status_fighting()
choose_magic = int(input('-----使用魔法/普通攻击(-1普通攻击,0/1/2..魔法攻击,查看状态-2):'))
IS_DEAD = self.attack_one_time(NPC_, choose_magic)
IS_DEAD = IS_DEAD * 2 # 用乘法,让我们不改变0,非0状态的情况下,标记是谁赢了
else:
choose_magic = random.choice(NPC_.HAS_MAGIC + [-1])
IS_DEAD = NPC_.attack_one_time(self, choose_magic)
IS_DEAD = IS_DEAD * 4 # 用乘法,让我们不改变0,非0状态的情况下,标记是谁赢了
if IS_DEAD >= 1:
print('【战斗结束】')
if IS_DEAD == 2:
print('【', self.NAME + '战胜了' + NPC_.NAME, '】')
elif IS_DEAD == 4:
print('【', NPC_.NAME + '战胜了' + self.NAME, '】')
return IS_DEAD
else:
if ROUND % 2 == 1:
print('【第' + str(1 + int(ROUND / 2)) + '回合结束!】\n===================================')
class World: # 世界类
# 剧本:遇到学姐
# 玩家初始化
PLAYER = Body()
# NPC初始化(先创造三个NPC,学姐,冰怪,火怪
NPCs = {
'school elder sister': Body(NAME_='school elder sister', HAS_MAGIC_=[0, 1, 2]),
'ice monster': Body(NAME_='ice monster', HAS_MAGIC_=[2]),
'fire monster': Body(NAME_='fire monster', HAS_MAGIC_=[0])
}
# 初始化世界:
def init(self):
player_name = input('请输入玩家名:')
self.PLAYER.NAME = player_name
print('....南理文字RPG0.2的世界开始了.....')
welcome_script = 'Hi,' + self.PLAYER.NAME + ',欢迎加入南理文字RPG0.2 '
# self.display_a_line(script=welcome_script)
print(welcome_script)
self.PLAYER.display_status()
self.PLAYER.talk_to(self.NPCs['school elder sister'],
SCRIPT_NPC_SCHOOL_SISTER) # 通过NPCs集合,调取'school elder sister'NPC
self.PLAYER.learn_magic_from(self.NPCs['school elder sister'])
def run(self):
self.PLAYER.fight_with(self.NPCs['fire monster'])
def image_on_image(img1, img2, x=100, y=100):
# 代表图像像素有多少行多少列多少通道数
rows, cols, channels = img2.shape # 获取图像2的shape信息 存入rows,cols,channels
roi = img1[0:rows, 0:cols] # 把背景图片img1的左上角按图像2的形状赋值给roi
img1[0 + y:rows + y, 0 + x:cols + x] = img2
cv2.imshow('res', img1)
return img1
# 当前状态、之前状态
cur_flag = -1
pre_flag = -1
# 无限循环
x, y = 100, 100 # 主角的坐标位置
delta = 15 # 主角每次走一步,走的步伐
while True:
# 每一帧
background = img1.copy() # 背景图像值拷贝
actor = img2.copy() # 主角图像值拷贝
# 获取键盘事件
flag = cv2.waitKey(0) # 程序等待用户敲击键盘,然后返回键盘值给flag
# Esc,退出
if flag == 27:
break
# 判断是否按下其他键
if flag > -1 and flag != pre_flag:
cur_flag = flag
pre_flag = flag
# 响应 actor移动事件
if cur_flag == ord('w'):
y = y - delta
elif cur_flag == ord('s'):
y = y + delta
elif cur_flag == ord('a'):
x = x - delta
elif cur_flag == ord('d'):
x = x + delta
else:
1 + 1
if x < 10:
x = 0
if y < 10:
y = 0
if x > 1435:
x = 1435
if y > 745:
y = 745
# 显示
print('Actor(x,y)=(' + str(x) + ',' + str(y) + ')')
background = image_on_image(background, actor, x, y) # 把背景图像,主角图像,x,y坐标传入,返回获取图像叠加的一个图像
cv2.destroyAllWindows()
# 显示一句话:两种方式,直接显示0,以及逐步显示1
def display_a_line(way, script='hello'):
for i in range(len(script)):
time.sleep(0.3)
print('\r', script[0:i], end='')
magic_world = World()
magic_world.init()
magic_world.run()