2023-07-29 09:53:44 +08:00
|
|
|
|
import cv2
|
|
|
|
|
|
import random
|
|
|
|
|
|
import warnings
|
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
from PIL import Image
|
|
|
|
|
|
|
|
|
|
|
|
|
2023-08-17 22:37:50 +08:00
|
|
|
|
def pix_random_change(img):
|
2023-07-29 09:53:44 +08:00
|
|
|
|
# Image转cv2
|
|
|
|
|
|
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
|
|
|
|
|
|
img[0, 0, 0] = random.randint(0, 0xfffffff)
|
|
|
|
|
|
# cv2转Image
|
|
|
|
|
|
img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
|
|
|
|
|
|
return img
|
|
|
|
|
|
|
|
|
|
|
|
|
2023-08-17 22:37:50 +08:00
|
|
|
|
def pix_random_change_file(path: str):
|
2023-07-29 09:53:44 +08:00
|
|
|
|
# 注意：cv2.imread()不支持路径中文
|
|
|
|
|
|
warnings.filterwarnings("ignore", category=Warning)
|
|
|
|
|
|
img = cv2.imread(path)
|
|
|
|
|
|
img[0, 0, 0] = random.randint(0, 0xfffffff)
|
|
|
|
|
|
cv2.imwrite(path, img)
|
|
|
|
|
|
|
