[英]Random flipping and RGB jittering/slight value change of image?
我想實現一個小型程序,該程序將隨機翻轉並引入RGB抖動/輕微值變化。
並盡可能將抖動/輕微值更改為彩色圖像中3層中的2層。
import cv2
import random
probofflip=0.5
probofRGBjit= 0.6
img=cv2.imread('path/to/img.png',1)
if (random.uniform(0,1)>1-probofflip):
img= cv2.flip(img,1)
if if (random.uniform(0,1)>1-probofRGBjit):
#function to jitter the RGB layers
#do something with resultant image.
如果您使用的是cv2,那么numpy對於此類操作非常有幫助。 抖動是指移動一些像素嗎? 本示例僅處理微小的價值變化。
import cv2
import numpy as np
from pylab import *
img = cv2.imread( r'C:/Users/Public/Pictures/Sample Pictures/Penguins.jpg' )
img = cv2.cvtColor(img, cv2.cv.CV_BGR2RGB) # cv2 defaul color code is BGR
h,w,c = img.shape # (768, 1024, 3)
noise = np.random.randint(0,50,(h, w)) # design jitter/noise here
zitter = np.zeros_like(img)
zitter[:,:,1] = noise
noise_added = cv2.add(img, zitter)
combined = np.vstack((img[:h/2,:,:], noise_added[h/2:,:,:]))
imshow(combined, interpolation='none')
如果要將每個顏色通道移動一些像素,則可以使用np.roll。 例如:
# shift each channel by 10 pixels
R = img[:,:,0]
G = img[:,:,1]
B = img[:,:,2]
RGBshifted = np.dstack( (
np.roll(R, 10, axis=0),
np.roll(G, 10, axis=1),
np.roll(B, -10, axis=0)
))
imshow(RGBshifted)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.