search

schardong / ifmorph

Implementation of "Neural Implicit Morphing of Face Images", published on CVPR 2024
https://schardong.github.io/ifmorph/
MIT License
13 stars 1 forks source link

Warp training using discrete images in place of continuous ones. #2

Closed schardong closed 5 months ago

schardong commented 5 months ago

Created a new script to train a warping warp-train-image.py. Most of the changes were in ifmorph.dataset. Created a new dataset for warping using discrete images (DiscreteImageWarpingDataset). Significant changes to ImageDataset as well, finally added batching code to the image dataset. Passing the pixel indices to ImageDataset.__getitem__ and returning the indices as well to propagate them to other instances (not used anymore, but kept the change).