Using this for a custom task/dataset

xh-liu / CC-FPSE

Code for NeurIPS 2019 paper "Learning to Predict Layout-to-image Conditional Convolutions for Semantic Image Synthesis"

128 stars 15 forks source link

Using this for a custom task/dataset #5

Closed balag59 closed 3 years ago

balag59 commented 4 years ago

Hi, Thank you for your amazing work! I'm working on an image to image translation task, from sketches of faces to their corresponding photos. Would this work well on this task and type of data? If yes, how would I go about training with a custom dataset? Thanks!

xh-liu commented 4 years ago

Hi, To adapt this approach to your task, you need to write your custom dataloader in folder "data/", and change necessary hyperparams (e.g., the dimension of the input label map).