Queries

[abhigoku10]

@fnzhan hi thanks for open-sourcing the code base , its really great work i have few queries

1. can we train the code for other semantic datasets like bdd100k / cityscapes? if so what changes have to be made
2. can we train the code for custom fashion dataset for region wise dressing ? if so what is the procedure

Thanks in advance

[fnzhan]

Hi,

1. of course, the model can be applied to other datasets. You should add the python file of the corresponding dataset, e.g., data/ade20k_dataset.py, and modify the processing code here

   UNITE/models/pix2pix_model.py

   Line 149 in d84b833

   def preprocess_input(self, data):

2. The model can work if a semantic correspondence can be built. Thus, if the region is annotated with semantic segmentation, the model can be applied by following similar procedure with ADE20K.