用Python實(shí)現(xiàn)屏幕截圖詳解

更新時(shí)間：2022年01月21日 15:22:52 作者：風(fēng)吹落葉花飄蕩

大家好，本篇文章主要講的是用Python實(shí)現(xiàn)屏幕截圖詳解，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話記得收藏一下

一、序言

在csdn上有很多純copy，這給我復(fù)現(xiàn)帶來(lái)了很大的麻煩，所以我想根據(jù)我的個(gè)人找截圖的經(jīng)歷記錄下來(lái)，給未來(lái)的自己看，免得忘記了云云。
由于我比較喜歡用opencv處理圖像，所以截屏最后都會(huì)附帶一個(gè)使用opencv顯示圖片的一個(gè)步驟。

二、環(huán)境配置

1、下載pyautogui包

pip install pyautogui -i https://pypi.tuna.tsinghua.edu.cn/simple/

注：使用pyautogui方法獲取截屏的話，只下載這個(gè)就行了

2、下載opencv-python包

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/

注：方便實(shí)現(xiàn)對(duì)圖片的進(jìn)一步處理，比如實(shí)時(shí)獲取云云。

3、下載PyQt5包

pip install PyQt5 -i https://pypi.tuna.tsinghua.edu.cn/simple/

注：使用PyQt方法截屏下載介個(gè)

4、下載pypiwin32包

pip install pypiwin32 -i https://pypi.tuna.tsinghua.edu.cn/simple/

注：使用PyQt方法截屏下載介個(gè)

三、屏幕截屏源碼與解析

1、使用pyautogui方法實(shí)現(xiàn)截屏

import pyautogui
import cv2
import numpy as np

img = pyautogui.screenshot(region=[300,50, 200, 100])  # 分別代表：左上角坐標(biāo)，寬高
#對(duì)獲取的圖片轉(zhuǎn)換成二維矩陣形式，后再將RGB轉(zhuǎn)成BGR
#因?yàn)閕mshow,默認(rèn)通道順序是BGR，而pyautogui默認(rèn)是RGB所以要轉(zhuǎn)換一下，不然會(huì)有點(diǎn)問(wèn)題
img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)

cv2.imshow("截屏",img)
cv2.waitKey(0)

優(yōu)點(diǎn)：

方便快捷，容易寫(xiě)核心部分就一行.速度快0.04s左右，基本可以達(dá)到實(shí)時(shí)截屏的效果?？梢宰杂纱_定截屏區(qū)域

缺點(diǎn)：

但是不能指定獲取程序的窗口，因此窗口也不能遮擋。 2、使用PyQt方法實(shí)現(xiàn)截屏 a.獲取窗口的句柄,也就是目標(biāo)窗口名title。

import win32gui

hwnd_title = dict() #創(chuàng)建字典保存窗口的句柄與名稱(chēng)映射關(guān)系


def get_all_hwnd(hwnd, mouse):
    if win32gui.IsWindow(hwnd) and win32gui.IsWindowEnabled(hwnd) and win32gui.IsWindowVisible(hwnd):
        hwnd_title.update({hwnd: win32gui.GetWindowText(hwnd)})

win32gui.EnumWindows(get_all_hwnd, 0)

for h, t in hwnd_title.items():
    if t!= "":
        print(h, t)

注：程序會(huì)打印所有窗口的hwnd和title，有了title就可以進(jìn)行截圖了。

b.使用PyQt5截屏核心程序

from PyQt5.QtWidgets import QApplication
import win32gui
import sys
#這個(gè)是截取全屏的
hwnd = win32gui.FindWindow(None, 'C:/Windows/system32/cmd.exe')
app = QApplication(sys.argv)
screen = QApplication.primaryScreen()
img = screen.grabWindow(hwnd).toImage()
img.save("screenshot.jpg")

注：如果想截取特定的窗口，只需要將C:/Windows/system32/cmd.exe換成上一個(gè)程序中打印的title，并且保證那個(gè)窗口沒(méi)有被你最小化即可

優(yōu)點(diǎn)：

方便快捷，容易寫(xiě)核心部分就一行.速度快0.04s左右，基本可以達(dá)到實(shí)時(shí)截屏的效果?？梢宰杂纱_定要截屏的窗口

缺點(diǎn)：

不可以自由確定截屏區(qū)域 c.使用PyQt5截屏用Mat格式顯示的核心程序

def convertQImageToMat(incomingImage):
    '''  Converts a QImage into an opencv MAT format  '''
    # Format_RGB32 = 4,存入格式為B,G,R,A 對(duì)應(yīng) 0,1,2,3
    # RGB32圖像每個(gè)像素用32比特位表示，占4個(gè)字節(jié)，
    # R，G，B分量分別用8個(gè)bit表示，存儲(chǔ)順序?yàn)锽，G，R，最后8個(gè)字節(jié)保留
    incomingImage = incomingImage.convertToFormat(4)
    width = incomingImage.width()
    height = incomingImage.height()

    ptr = incomingImage.bits()
    ptr.setsize(incomingImage.byteCount())
    arr = np.array(ptr).reshape(height, width, 4)  # Copies the data
    # arr為BGRA，4通道圖片
    return arr

from PyQt5.QtWidgets import QApplication
import win32gui
import sys
import cv2
import numpy as np
hwnd = win32gui.FindWindow(None, '劍士之魂中文版小游戲,在線玩,4399小游戲 - 360安全瀏覽器 13.1')
app = QApplication(sys.argv)
screen = QApplication.primaryScreen()
img = screen.grabWindow(hwnd).toImage()

img=convertQImageToMat(img)#將獲取的圖像從QImage轉(zhuǎn)換為RBG格式
cv2.imshow("asd",img)      #imshow
cv2.waitKey(0)