pix2vertex.pytorch
pix2vertex.pytorch copied to clipboard
pix2pix network
Hi, this is a great implementation! not only the code is complete from model to plotting phase, but the 3D plot is detailed as well!
As a new deep learning enthusiast I have learned alot from your code, but I still don't understand about the pix2pix output shape : (1, 7, 512, 512)
from reconstructor.py I just noticed that the 7 is separated into two variables: im_pncc and im_depth,
but I have difficulties understanding what im_pncc is and why in the following code:
mask = np.any(im_depth, axis=2) * np.all(im_pncc, axis=2)
is not just:
mask = np.any(im_depth, axis=2)
since im_depth across each of 512x512 already shows the location of the face, therefore there is no need for im_pncc (?)
and lastly, is there any way to train the pix2pix model for full-body 3D?
Thank you !