Utiliser numpy pour masquer une image avec un motif?

Je suis en utilisant numpy pour construire pixel tableaux. Une image en 800x600 est un 3-dimensions tableau de uint8, 800x600x3. J'ai aussi un tableau similaire avec un motif fixe (un damier, voir ici). J'ai un autre tableau, 800x600 des valeurs de masque. Lorsque le masque est égale à zéro, je veux copier le motif pixel de l'image en pixels. L'endroit où le masque n'est pas zéro, je veux quitter le pixel de l'image seule.

>>> image.shape
(800, 600, 3)
>>> chex.shape
(800, 600, 3)
>>> mask.shape
(800, 600)

Cela se sent comme il devrait fonctionner:

image[mask == 0,...] = chex

mais donne "ValueError: tableau n'est pas broadcastable pour corriger la forme".

Que dois-je utiliser pour copier chex les pixels en pixels de l'image où le masque est égale à zéro?

OriginalL'auteur Ned Batchelder | 2010-01-30

  1. 14
    idx=(mask==0)
image[idx]=chex[idx]

    Noter que image a la forme (800,600,3), tandis que idx a la forme (800,600). Les règles pour l'indexation de l'état

    si la sélection d'un tuple est plus petit que
    n, alors que de nombreux : les objets nécessaires
    sont ajoutés à la fin de la sélection
    tuple de sorte que la modification de la sélection
    tuple est de longueur N.

    Donc l'indexation des tableaux ont une sorte de capacité de diffusion de leur propre. idx's de la forme devient promu (800,600,:)

    OriginalL'auteur unutbu

  2. 1

    Je voulais illustrer un exemple d'utilisation de @unutbu réponse. Dans ce scénario, j'ai une image d'un chat que j'ai tourner. Cette rotation provoque des bords noirs que moche, surtout lorsqu'il est collé sur un non-arrière-plan noir. 

    import matplotlib.pyplot as plt
from scipy.ndimage import rotate


cat = plt.imread('cat.jpeg')
bg = plt.imread('background.jpeg')


rotcat = rotate(cat, angle=8, reshape=True) ## rotating creates some black edges
height, width, _ = rotcat.shape

bgcopy = bg.copy() ## create a copy of the background; paste on copy

x, y = 40, 50 
bgcopy[x:x+height, y:y+width] = rotcat
plt.imsave('cat-on-bg-mask.jpg', bgcopy)

    Utiliser numpy pour masquer une image avec un motif?

    Je trouve que les zones du masque et de remplacer ces valeurs avec l'arrière-plan d'origine des valeurs

    mask_ind = (bgcopy == 0)
bgcopy[mask_ind] = bg[mask_ind]
plt.imsave('cat-on-bg.jpg', bgcopy)

    Utiliser numpy pour masquer une image avec un motif?

    Je tiens également à noter que PIL.Image (à partir de la Pillow bibliothèque) a la possibilité de coller une image sur une autre image avec moins d'étapes.

    OriginalL'auteur Jon

  3. 0

    Essayer:

    image[mask[:] == 0,...] = chex[mask[:] == 0,...]
    "ValueError: tableau n'est pas broadcastable pour corriger la forme"

    OriginalL'auteur ezod

  4. 0

    J'ai utilisé des tableaux de 8x6x3, 8x6x3, et 8x6 représenter votre image, tableau, vérificateur de pile, puis le masque de tableau, respectivement.

    # first create mini-versions of your arrays:
mask = NP.random.random_integers(0, 1, 48).reshape(8, 6)
img = NP.random.random_integers(3, 9, 8*6*3).reshape(8, 6, 3)
chk = NP.ones((8, 6, 3))

# all the work done in these two lines
mask = mask[:,:,NP.newaxis]
res = NP.where(mask==0, chk, img)
    Les formes des trois images, mask.shape (8, 6), chk.shape (8, 6, 3) and img.shape (8, 6, 3) ne semble pas correspondre à l'exigence de np.where.

    OriginalL'auteur doug