search

ericwu09 / mammo-cigan

Code for "Conditional infilling GAN for Data Augmentation in Mammogram Classification"
9 stars 3 forks source link

Train and Inference on my own dataset #1

Open lmz123321 opened 4 years ago

lmz123321 commented 4 years ago

Hi,Thanks for your code! I am currently trying to use ciGAN to do some data augmentation on my own dataset. Your README is pretty simple, so I have some question: (1)How should I prepare my data? What the folder structures are like? (2)Where should I put the VGG-19 pretrained model? (3)What the difference between '--val' and '--synthetic'? If convenient, Could you provide more detailed piplines to re-produce your work? Greatful for your help!

viniandrd commented 4 years ago

I'm gonna be so thankful if @ericwu09 answer this question.. I'm trying to run his code too but I'm not succeeding... Need more details about what tensorflow version he used and how should I prepare the dataset..

viniandrd commented 4 years ago

Btw, I'm running on PyCharm with Tensorflow GPU 1.15.