mask from zbuffer

[wangxinjun56]

tri_map_2d, mask_i = ZBuffer_Rendering_CUDA_op_v2_sz224(vertex2d_i, tri_const, visible_tri_i)

the mask_i is 224*224 , shape is (-1,);

masks.append(mask_i)
masks = tf.stack(masks)

return images, masks

where the mask shape is wrong ,that

G_images_300W_mask = tf.multiply(input_masks_300W, tf.expand_dims(G_images_300W_mask, -1))
G_images_300W = tf.multiply(G_images_300W, G_images_300W_mask) + tf.multiply(input_images_300W, 1 - G_images_300W_mask)

[tranluan]

Can you elaborate on your question?

[wangxinjun56]

the G_images_300W_mask .shape is wrong, should be batchsize * 224 * 224 * 3 that can multiply with input_masks_300W

[tranluan]

The shape is batchsize * 224 * 224 * 1, which is ok for the multiplication.

[linshuheng6]

@tranluan Hello， could you please tell how to render the generated image with only face part? Or how to obtain the index of those vertices of face which doesn't include the neck and ear? The mask from Z-buffer includes neck, which is not expected. Thank you !