eliahuhorwitz / DeepSIM

Official PyTorch implementation of the paper: "DeepSIM: Image Shape Manipulation from a Single Augmented Training Sample" (ICCV 2021 Oral)

Home Page:http://www.vision.huji.ac.il/deepsim

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Train on dataset with several samples

andorxornot opened this issue · comments

commented

What do you think i need to change to train with a dataset consisting of several samples?

This project does not support multiple images, however, it can be easily modified to support. To train on multiple images you may refer to the original pix2pixHD as our architecture is based on it, in addition, you may find data/base_dataset.py useful. Keep in mind performance may differ as a result of adding additional images.

commented

thanks!