Сохранять пиксели на изображении, которые не являются черными - PullRequest
Новая
       22

Сохранять пиксели на изображении, которые не являются черными

0 голосов
/ 30 апреля 2018

Я хочу обработать цветное изображение и сохранить только пиксели, которые не являются черными. Я написал некоторый код, но он, кажется, не делает то, что, я думаю, делает: 

import cv2
import numpy as np
import tensorly as tl

def eq_svtn(T, b, M, Omega):
X = 0
for i in range(3):
    X += b[i]*tl.fold(M[i],i,T.shape)
X = X/np.sum(b)

for i in range(T.shape[0]):
    for j in range(T.shape[1]):
        if Omega[i, j, :].all() != 0:
            X[i, j, :] = T[i, j, :]
return X

def silrtc(X,b,K,O):
a = np.random.dirichlet(np.ones(3),size=1)
a = a[0,:]
M = [0, 0, 0]
for i in range(K):
    X1 = tl.unfold(X, 0)
    X2 = tl.unfold(X, 1)
    X3 = tl.unfold(X, 2)
    u1, s1, v1 = LA.svd(X1, full_matrices=False)
    u2, s2, v2 = LA.svd(X2, full_matrices=False)
    u3, s3, v3 = LA.svd(X3, full_matrices=False)
    U = [u1,u2,u3]
    V = [v1,v2,v3]
    S = [s1,s2,s3]
    for j in range (3):
        M[j] = shrinkage_op(U[j],V[j],S[j],a[j]/b[j],X.shape)
    X = eq_svtn(X,b,M,O)
return X


im = cv2.imread('target.jpg')
original_shape = im.shape
zzo = np.zeros(im.shape)
shp = (int(original_shape[0]),int(mt.floor(0.9*original_shape[1])),int(original_shape[2]))
shp2 = (int(original_shape[0]),int(mt.ceil(0.1*original_shape[1])),int(original_shape[2]))
zz = np.zeros(shp)
oo = 0xFF*np.ones(shp2)
cct = np.concatenate((zz,oo),1)
np.random.shuffle(cct.reshape(-1, 3))
cct = cct.astype(np.uint8)
im2 = np.bitwise_and(im,cct)
b = [0.0001, 0.0001,0.0001]
xx = silrtc(im2,b,64,cct)

Черный пиксель равен [0,0,0], b - это массив из 3-х чисел, M - это набор двумерных массивов (матриц), T - это изображение (3-D). массив (тензор) с черными пикселями, X - цветное изображение с удаленными черными пикселями.

...