| import numpy as np |
| import cv2 |
| from PIL import Image |
|
|
|
|
| def pil_to_cv(img): |
| new_img = np.array(img, dtype=np.uint8) |
| if new_img.ndim == 2: |
| pass |
| elif new_img.shape[2] == 3: |
| new_img = cv2.cvtColor(new_img, cv2.COLOR_RGB2BGR) |
| elif new_img.shape[2] == 4: |
| new_img = cv2.cvtColor(new_img, cv2.COLOR_RGBA2BGRA) |
| return new_img |
|
|
| def cv_to_pil(img): |
| new_img = img.copy() |
| if new_img.ndim == 2: |
| pass |
| elif new_img.shape[2] == 3: |
| new_img = cv2.cvtColor(new_img, cv2.COLOR_BGR2RGB) |
| elif new_img.shape[2] == 4: |
| new_img = cv2.cvtColor(new_img, cv2.COLOR_BGRA2RGBA) |
| new_img = Image.fromarray(new_img) |
|
|
| return new_img |
|
|
