Python-OpenCV讀取與保存圖片

1. OpenCV讀取圖片

1.1 簡述

　　OpenCV讀取圖片的方法是cv2.imread()，讀取出來圖片的格式是BGR與常規的彩色圖像的格式（RGB）相反，這一點一定要注意。

　　OpenCV顯示圖片的方法是cv2.imshow(),顯示的格式是BGR。

　　小生就言於此O(∩_∩)O哈哈~，直接上例子。

1.2 例子

1.2.1 Code

import cv2


def cv2_imread():
    # 圖片路徑，相對路徑
    image_path = "./fusion_datasets/1.jpg"
    # 讀取圖片,格式為BGR
    image = cv2.imread(image_path)
    # 顯示圖片形狀
    print("image_shape: ", image.shape)
    # 縮放圖片
    width = int(image.shape[0] / 2)
    height = int(image.shape[1] / 2)
    image = cv2.resize(image, (height, width), interpolation=cv2.INTER_CUBIC)
    # 顯示圖片
    cv2.imshow('girl', image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()


if __name__ == '__main__':
    cv2_imread()

 

1.2.2 結果顯示

 

 

 

2. OpenCV保存圖片

2.1 簡述

 　　OpenCV保存圖片用cv.imwrite(filename, img)，filename為保存的路徑名稱，img為要保存的圖片。

　　但實際應用中不會這么簡單，(*^▽^*)我掐指一算，絕了O(∩_∩)O哈哈~，路過的寶寶們大多數的是想將tensor（tensor先轉為numpy）或numpy轉為cv2保存為圖片。

　　需要注意如下幾點：

　　　　1. cv2保存的共三個維度（高，寬，通道）。

　　　　2. 保存的圖片格式也是BGR。

　　　　3. 將numpy轉為np.uint8格式

2.2 例子

2.2.1 Code

import cv2
import numpy as np
import torchvision.transforms as transforms


def cv2_imwrite():
    # =============== 模擬圖片為tensor數據 ============ #
    # 圖片路徑，相對路徑
    image_path = "./fusion_datasets/1.jpg"
    # 讀取圖片， 圖片格式為（height, width, channel）
    image = cv2.imread(image_path)
    # 圖像轉為tensor，此時圖片變為（channel, height, width），並將0-255歸一化為0-， 減小計算
    input_transform = transforms.Compose([
        transforms.ToTensor(),
    ])
    # unsqueeze(0)：是在擴展維度由三維擴展為4維，(bathch_size, channel, height, width)
    # 這是實際實驗中可能遇到，所以我才擴展維度的，也可以不擴展
    image = input_transform(image).unsqueeze(0)

    # =============== 數據准備完成，開始保存 ============ #
    # 1.首先將tensor降維度，4維降為3維:image.squeeze(0)
    # 2.tensor轉為numpy:image.squeeze(0).numpy()
    # 3.轉換維度，由（channel, height, width）變為（height, width, channel）
    # 4.數據類型由numpy轉換維np.uint8（這是圖片應有的格式）
    # 5.保存圖片，執行函數cv2.imwrite(filename, img),filename:保存路徑名要, img:保存的圖片
    image = np.transpose(image.squeeze(0).numpy() * 255.0, (1, 2, 0)).astype(np.uint8)
    cv2.imwrite("./results/" + "girl.jpg", image)


if __name__ == '__main__':
    cv2_imwrite()

 

2.2.2 結果顯示

　　圖片已保存~~

         

　　  打開圖片查看~~，哇哦，so beautiful(*^▽^*)！

 　　

 

 3. 總結

　　努力去愛周圍的每一個人，付出，不一定有收獲，但是不付出就一定沒有收獲！ 給街頭賣藝的人零錢，不和深夜還在擺攤的小販討價還價。願我的博客對你有所幫助(*^▽^*)(*^▽^*)！