目錄
- 前言
- 一、核心功能設(shè)計
- 二、實現(xiàn)步驟
- 1. 解析提取，修改圖片
- 2.初始化動畫掛件
- 3.動畫掛件功能實現(xiàn)
- 4.打包配置
前言
前段時間，寫了篇博客自己用python做的一款超炫酷音樂播放器。有粉絲問我，音樂播放器為什么要用PyQt5，效果是不是比Tkinter贊？PyQt5真的可以實現(xiàn)這些炫酷的UI畫面嗎？之前沒接觸過PyQt5，能不能多分享一些這方面的開發(fā)案例？
今天就帶大家，一起用Python的PyQt5開發(fā)一個有趣的自定義桌面動畫掛件，看看實現(xiàn)的動畫掛件效果！
下面，我們開始介紹這個自定義桌面動畫掛件的制作過程。
一、核心功能設(shè)計
總體來說，我們需要實現(xiàn)將自己喜歡的動態(tài)圖gif或者視頻轉(zhuǎn)成一個桌面動畫掛件，知識點主要包含了對GIF圖/視頻解析，人像提取分割，PyQt5窗體設(shè)置，自定義掛件動畫實現(xiàn)，ico圖標生成，程序打包等。
拆解需求，大致可以整理出我們需要分為以下幾步完成：
- 對gif或者視頻進行逐幀解析，獲取轉(zhuǎn)換的圖片，提取圖像中人體區(qū)域，并對圖片進行批量尺寸大小修改替換
- 初始化設(shè)置動畫掛件窗體顯示效果，窗體位置、大小等
- 桌面動畫掛件功能實現(xiàn)，動畫輪播、鼠標控制掛件位置拖動
- 掛件打包圖標設(shè)置、打包配置
二、實現(xiàn)步驟
1. 解析提取，修改圖片
GIF圖解析：
Gif動態(tài)圖資源大家可以根據(jù)自己的喜好，自己選擇。博主就用之前寫過的仙女蹦迪動態(tài)Gif來演示效果。
首先我們需要將Gif動態(tài)圖按照每一幀進行解析 ，轉(zhuǎn)換成圖片格式。代碼如下：
from PIL import Image # 導入PIL的Image包
import os
gifFileName = "./demo.gif" # 把gif圖賦值給gifFileName
im = Image.open(gifFileName) # 使用Image的open函數(shù)打開test.gif圖像
pngDir = gifFileName[:-4] # 倒著從gifFileName中的倒數(shù)第四個開始取字符（跳過.gif），賦值給pngDir，作為文件夾的名字
if not os.path.exists(pngDir):
os.makedirs('./img') # 用圖片名創(chuàng)建一個文件夾，用來存放每幀圖片，名字為pngDir的值
try:
while True: # 死循環(huán)
current = im.tell() # 用tell函數(shù)保存當前幀圖片，賦值給current
im.save(pngDir+'/'+str(current+1)+'.png') # 調(diào)用save函數(shù)保存該幀圖片
im.seek(current+1) # 調(diào)用seek函數(shù)獲取下一幀圖片，參數(shù)變?yōu)閏urrent幀圖片+1
# 這里再次進入循環(huán)，當為最后一幀圖片時，seek會拋出異常，代碼執(zhí)行except
except EOFError:
pass # 最后一幀時，seek拋出異常，進入這里，pass跳過
這樣就可以把動態(tài)Gif圖轉(zhuǎn)換成圖片了，效果如下:
視頻解析：
同理，對視頻解析，也是按照每一幀進行解析，轉(zhuǎn)換成圖片格式。核心代碼如下：
# 將視頻按照每一幀轉(zhuǎn)成圖片png
import cv2
videoFileName = "./demo.mp4" # 把視頻路徑賦值給videoFileName
pngDir = videoFileName[:-4] # 倒著從gifFileName中的倒數(shù)第四個開始取字符（跳過.后綴），賦值給pngDir，作為文件夾的名字
if not os.path.exists(pngDir):
os.makedirs(pngDir) # 用圖片名創(chuàng)建一個文件夾，用來存放每幀圖片，名字為pngDir的值
# 視頻處理 分割成一幀幀圖片
cap = cv2.VideoCapture(videoFileName)
num = 1
while True:
# 逐幀讀取視頻 按順序保存到本地文件夾
ret, frame = cap.read()
if ret:
cv2.imwrite(f"{pngDir}/{num}.png", frame) # 保存一幀幀的圖片
num += 1
else:
break
cap.release() # 釋放資源
效果如下：
逐幀提取的圖片已經(jīng)拿到了，下面我們需要對這些圖片中的人像進行分割提取。
人像分割：
我們調(diào)用的是百度開放的人體分析接口 – 百度AI開放平臺鏈接。
這里面我們可以創(chuàng)建一個人像分割的應用，其中的API Key及Secret Key后面我們調(diào)用人臉識別檢測接口時會用到。
我們可以看到官方提供的幫助文檔，介紹地很詳細。如何調(diào)用請求URL數(shù)據(jù)格式，向API服務地址使用POST發(fā)送請求，必須在URL中帶上參數(shù)access_token，可通過后臺的API Key和Secret Key生成。這里面的API Key和Secret Key就是我們上面提到的。
那我們?nèi)绾潍@取空背景的人像圖片呢？根據(jù)API文檔，可以看到里面有個type屬性設(shè)置為foreground 就可以提取空背景的人像圖片。
人像分割的接口流程基本就已經(jīng)清楚了，可以進行代碼實現(xiàn)了。
# 保存圖片
def save_base_image(img_str, filename):
img_data = base64.b64decode(img_str)
with open(filename, 'wb') as f:
f.write(img_data)
# 獲取token
def get_token():
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentialsclient_id=' + client_id + 'client_secret=' + client_secret
request = urllib.request.Request(host)
request.add_header('Content-Type', 'application/json; charset=UTF-8')
response = urllib.request.urlopen(request)
token_content = response.read()
if token_content:
token_info = json.loads(token_content)
token_key = token_info['access_token']
return token_key
# 人像分割
def body_seg_fore(filename, resultfilename):
request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"
# 二進制方式打開圖片文件
f = open(filename, 'rb')
img = base64.b64encode(f.read())
params = dict()
params['image'] = img
params['type'] = 'foreground'
params = urllib.parse.urlencode(params).encode("utf-8")
# params = json.dumps(params).encode('utf-8')
access_token = get_token()
request_url = request_url + "?access_token=" + access_token
request = urllib.request.Request(url=request_url, data=params)
request.add_header('Content-Type', 'application/x-www-form-urlencoded')
response = urllib.request.urlopen(request)
content = response.read()
if content:
# print(content)
content = content.decode('utf-8')
# print(content)
data = json.loads(content)
# print(data)
img_str = data['foreground']
save_base_image(img_str, resultfilename)
這樣我們就可以根據(jù)圖片，進行人像分割，提取出空背景的人像圖。效果如下：
如果提取的人像圖片大小不符合我們的要求，我們還可以對圖片大小進行調(diào)整。
file_list = os.listdir("./image") # 讀取當前文件夾所有文件
# print(file_list)
n = len(file_list)
for i in range(n):
s = str(file_list[i])
if s[-4:] == ".png": # 檢查后綴
src = os.path.join(os.path.abspath('./image/'), s) # 原先的圖片名字
img = Image.open(src)
new_img = img.resize((128, 128), Image.BILINEAR)
new_img.save(src)
我們需要的空背景圖片已經(jīng)拿到了，接下來我們來實現(xiàn)桌面掛件功能。
2.初始化動畫掛件
# 窗體初始化
def windowinit(self):
self.x = 1650
self.y = 860
self.setGeometry(self.x, self.y, 300, 300)
self.setWindowTitle('My Gadgets')
self.img_num = 1
self.img_path = './image/{file}/{img}.png'.format(file=self.dis_file, img=str(self.img_num))
self.lab = QLabel(self)
self.qpixmap = QPixmap(self.img_path)
self.lab.setPixmap(self.qpixmap)
self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint | Qt.SubWindow)
self.setAutoFillBackground(False)
self.setAttribute(Qt.WA_TranslucentBackground, True)
self.show()
def __init__(self):
super(Gadgets, self).__init__()
self.dis_file = "img1"
self.windowinit()
self.icon_quit()
self.pos_first = self.pos()
self.img_count = len(os.listdir('./image/{}'.format(self.dis_file)))
這樣圖片掛件就可以顯示出來，效果如下：
現(xiàn)在的桌面掛件是靜態(tài)顯示，下面我們可以通過計時器進行掛件動畫輪播顯示。
3.動畫掛件功能實現(xiàn)
動畫輪播：
self.timer = QTimer()
self.timer.timeout.connect(self.img_update)
self.timer.start(100)
def img_update(self):
if self.img_num self.img_count:
self.img_num += 1
else:
self.img_num = 1
self.img_path = './image/{file}/{img}.png'.format(file=self.dis_file, img=str(self.img_num))
self.qpixmap = QPixmap(self.img_path)
self.lab.setPixmap(self.qpixmap)
鼠標控制掛件位置拖動：
def mousePressEvent(self, QMouseEvent):
if QMouseEvent.button() == Qt.LeftButton:
self.pos_first = QMouseEvent.globalPos() - self.pos()
QMouseEvent.accept()
self.setCursor(QCursor(Qt.OpenHandCursor))
def mouseMoveEvent(self, QMouseEvent):
if Qt.LeftButton:
self.move(QMouseEvent.globalPos() - self.pos_first)
print(self.pos())
self.x, self.y = self.pos().x, self.pos().y
QMouseEvent.accept()
def quit(self):
self.close()
sys.exit()
至此，自定義動畫掛件功能已經(jīng)實現(xiàn)完成，動畫掛件效果如下：
4.打包配置
前段時間有粉絲問我，Python能不能打包成exe？如何打包呢？今天就通過這個來一起介紹下。
Python常用的打包工具是第三方庫Pyinstaller，首先需要安裝下pyinstaller。
接下來，我們需要打開命令窗口，切換到項目目錄下再執(zhí)行打包命令。
pyinstaller -F -i ./img.ico Qt_Gadgets.py
打包常用的參數(shù)如下：
- -F 表示生成單個可執(zhí)行文件
- -w 表示去掉控制臺窗口，這在GUI界面時非常有用。不過如果是命令行程序的話那就把這個選項刪除吧！
- -p 表示你自己自定義需要加載的類路徑，一般情況下用不到
- -i 表示可執(zhí)行文件的圖標
之前有小伙伴問我，打包的圖標需要.ico后綴的圖片，應該如何把普通圖片轉(zhuǎn)成圖標格式.ico呢？Python當然可以幫你實現(xiàn)，今天就一起教給大家。核心代碼如下：
import PythonMagick
# 生成圖標ico(png格式圖片轉(zhuǎn)成ico)
img = PythonMagick.Image('./image/img1/1.png')
# 這里要設(shè)置一下尺寸，不然會報ico尺寸異常錯誤
img.sample('128x128')
img.write('./img.ico')
圖標得到了，下面我們就可以進行打包操作了。
打包完成之后，我們可以看見項目目錄下會有生成的exe程序。
至此，整個自定義動畫掛件就全部完成了，下面我們一起運行下exe看看動畫掛件效果。
今天我們就到這里，明天繼續(xù)努力！
如果本篇博客有任何錯誤，請批評指教，不勝感激 ！
到此這篇關(guān)于用Python做個個性的動畫掛件讓桌面不單調(diào)的文章就介紹到這了,更多相關(guān)Python自定義動畫掛件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- AI:如何訓練機器學習的模型
- 使用JavaSE來模擬斗地主
- 用python做個代碼版的小仙女蹦迪視頻
- 詳細過程帶你用Python做車牌自動識別系統(tǒng)
- Python做個自定義動態(tài)壁紙還可以放視頻
- 自己用python做的一款超炫酷音樂播放器
- 我用Python做個AI出牌器斗地主把把贏
