search

AntreasAntoniou / DAGAN

DAGAN: Data Augmentation Generative Adversarial Networks
https://arxiv.org/abs/1711.04340
MIT License
415 stars 103 forks source link

can I use DAGAN for binary classes #20

Open khatiakilanava opened 5 years ago

khatiakilanava commented 5 years ago

Hello, I really need your help, it will be very much appreciated as I can't solve it, unfortunately. I have a training dataset of classes 1 and 0. 6000x50x50x3 per each class. I want to generate more data for both classes. Can I do it using DAGAN? If yes I was wondering how to define this: " Then you need to choose which classes go to each of your training, validation and test sets".

In your code snippet from load_dataset will this be correct? x_train, x_test, x_val = self.x[:1], self.x[1:2], self.x[1:2] x_train = x_train[:gan_training_index] return x_train, x_test, x_val I guess here synthetic images for only one class will be generated, what should I do to generate for another class as well?

tobschu94 commented 5 years ago

Hello @khatiakilanava , did you solve your issue and can provide some help. I also have only two classes containing some images and have no idea how to split this dataset. Thanks