用Python實現屏幕截圖詳解

一、序言

在csdn上有很多純copy,這給我復現帶來瞭很大的麻煩,所以我想根據我的個人找截圖的經歷記錄下來,給未來的自己看,免得忘記瞭雲雲。
由於我比較喜歡用opencv處理圖像,所以截屏最後都會附帶一個使用opencv顯示圖片的一個步驟。

二、環境配置

1、下載pyautogui包

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

註:使用pyautogui方法獲取截屏的話,隻下載這個就行瞭

2、下載opencv-python包

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

註:方便實現對圖片的進一步處理,比如實時獲取雲雲。

3、下載PyQt5包

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

註:使用PyQt方法截屏下載介個

4、下載pypiwin32包

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

註:使用PyQt方法截屏下載介個

三、屏幕截屏源碼與解析

1、使用pyautogui方法實現截屏

import pyautogui
import cv2
import numpy as np

img = pyautogui.screenshot(region=[300,50, 200, 100])  # 分別代表:左上角坐標,寬高
#對獲取的圖片轉換成二維矩陣形式,後再將RGB轉成BGR
#因為imshow,默認通道順序是BGR,而pyautogui默認是RGB所以要轉換一下,不然會有點問題
img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)

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

優點:

方便快捷,容易寫核心部分就一行.速度快0.04s左右,基本可以達到實時截屏的效果。可以自由確定截屏區域

缺點:

但是不能指定獲取程序的窗口,因此窗口也不能遮擋。 2、使用PyQt方法實現截屏 a.獲取窗口的句柄,也就是目標窗口名title。

import win32gui

hwnd_title = dict() #創建字典保存窗口的句柄與名稱映射關系


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)

註:程序會打印所有窗口的hwnd和title,有瞭title就可以進行截圖瞭。

b.使用PyQt5截屏核心程序

from PyQt5.QtWidgets import QApplication
import win32gui
import sys
#這個是截取全屏的
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換成上一個程序中打印的title,並且保證那個窗口沒有被你最小化即可

優點:

方便快捷,容易寫核心部分就一行.速度快0.04s左右,基本可以達到實時截屏的效果。可以自由確定要截屏的窗口

缺點:

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

def convertQImageToMat(incomingImage):
    '''  Converts a QImage into an opencv MAT format  '''
    # Format_RGB32 = 4,存入格式為B,G,R,A 對應 0,1,2,3
    # RGB32圖像每個像素用32比特位表示,占4個字節,
    # R,G,B分量分別用8個bit表示,存儲順序為B,G,R,最後8個字節保留
    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轉換為RBG格式
cv2.imshow("asd",img)      #imshow
cv2.waitKey(0)

四、結語

我再查詢瞭多個截屏的方法之後,發現就學會這兩種截屏方法即可,所以就沒有再放其他方法瞭

到此這篇關於用Python實現屏幕截圖詳解的文章就介紹到這瞭,更多相關Python屏幕截圖內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: