1. 程式人生 > 實用技巧 >不到100行程式碼 Python製作一個九宮格圖片生成器,炫酷朋友圈!

不到100行程式碼 Python製作一個九宮格圖片生成器,炫酷朋友圈!

朋友圈下面的這種圖片排列風格,相比大家一定會很熟悉,有關於職位招聘的

祝賀節日的,

篩自己美照的,

這種因為圖片剛好為 3*3 的排列方式,所以被稱為 9 宮格圖片風格,圖片的生成原理就是把一張圖片按區域等比例分為 9 份碎片,朋友圈髮狀態時只需要自己調整一下圖片碎片放置位置即可。

雖然 9 宮格製作原理相對比較簡單,但它的加入讓朋友圈中只有一張圖片的動態在視覺方面提升了一個檔次

接下來 本文將介紹如何用 Python 將一張圖片轉化為 9 宮格,並加入 GUI 介面,封裝成一個程式,先看一下程式的預覽效果:

本次製作的環境配置介紹如下:

  • Python 3.7;
  • Opencv: 3.4;
  • PyQt5 : 5.9;

製作步驟

首先先將一下九宮格圖片轉換的基本思路:

  • 1,先把圖片變為正方形,不夠的邊用白色畫素填充;
  • 2,找到圖片的三等距離,用 for 遞迴形式 把圖片劃分為 9 個區域,並存儲為列表,因為這裡的圖片是以陣列形式表示,因此根據陣列形式很好劃分;
  • 3,把 2 分割得到的圖片列表分別進行儲存,利用檔名來進行編號;

核心程式碼如下:

 if self.open_file_path and self.save_file_path:
            try:
                img = cv2.imread(self.open_file_path)
                if len(img.shape) == 2: # 判斷是否為灰度圖
                    last_dim = 1
                else:
                    last_dim = 3
                if img.shape[0] != img.shape[1]:
                    # 長寬不一致
                    new_image = np.zeros((max(img.shape), max(img.shape), last_dim), dtype=np.uint8) + 255
                    # 影象填充
                    new_image[
                    int((new_image.shape[0] - img.shape[0]) / 2):img.shape[0] + int((new_image.shape[0] - img.shape[0]) / 2),
                    int((new_image.shape[1] - img.shape[1]) / 2):img.shape[1] + int((new_image.shape[1] - img.shape[1]) / 2),:] = img
                else:
                    new_image = img
                # 開始進行影象分割
                col_width = int(new_image.shape[0] / 3)

                # 得到九宮格影象
                image_list = [new_image[i * col_width:(i + 1) * col_width, j * (col_width):(j + 1) * col_width, :] for i in
                              range(3) for j in range(3)]
                for i in range(9):
                    image_name = str(i)
                    save_image_path = os.path.join(self.save_file_path, f'{image_name}.png')
                    cv2.imwrite(save_image_path, np.array(image_list[i]))
                    print(f'save {image_name} sucessfully!')
                QMessageBox.information(self,'info','轉換完成!')
            except Exception as e:
                print(e)
                QMessageBox.warning(self,'error',f'轉換失敗{str(e)}')
        else:
            QMessageBox.information(self,'err','檔案為空,請重新操作')

GUI 封裝

第二部分就是 GUI 封裝,這裡我用的是 PyQt5 ,建立一個 QWidgt 元件,放置三個按鈕,兩個 LineEdit 共5個元件,因為沒有設定任何屬性(顏色、背景、互動效果),也沒排版,所以程式比較簡陋

在程式裡,還在互動介面中加入了一些異常資訊處理,比如開啟地址為空、轉換失敗,程式會給相應的提示,比如下面這種,在沒有選擇資料夾時點選 開始轉換按鈕,給出了錯誤提示

GUI 介面的主要程式碼部分:

    def open_origin_file(self):
        open_file = QFileDialog.getOpenFileName(None,"Open File","C:/","Image (*.png)")
        if open_file[0]:
            print(open_file[0])
            self.open_file_path = open_file[0]
            self.line_edit.setText(self.open_file_path)
        else:
            QMessageBox.warning(self,"info","Fail to open file, please try it again!")

    def save_file(self):

        open_file = QFileDialog.getExistingDirectory(None,'Open File','C:/')
        if open_file:
            print(open_file)
            self.save_file_path = open_file
            self.line_edit1.setText(str(self.save_file_path))
        else:
            QMessageBox.warning(self,'info','Fail to open file, Please try it again!')

這裡我找了一張圖片來測試一下這個小程式,圖片是一個動漫角色-皮卡丘,效果如下,

最後隨意找了一張圖片做成九宮格發到了朋友圈,結果在下面第二張圖片(原圖為第一張),客觀一點地說,相對於單張圖片視覺效果並沒有提升,主要原因就是原圖的長寬高比例不一致,也就是說,對於某些長寬不一致的圖片在轉化為九宮格之前最好裁剪一下效果會更好,否則並不適合轉化為九宮格

原圖

沒裁剪直接轉化的圖

裁剪之後轉化的圖

裁剪之後在轉化視覺效果就有明顯的提升,感興趣的小夥伴們也來試一下,關於文章完整原始碼,公號 【Z先生點記】後臺回覆 九宮格即可獲取。

好了,以上就是本篇文章的全部內容啦,最後感謝大家的閱讀!