search

wchen342 / SketchyGAN

Code for paper "SketchyGAN: Towards Diverse and Realistic Sketch to Image Synthesis"
MIT License
131 stars 32 forks source link

How could I generate the .tfrecord myself? #8

Closed juefeiyuan closed 6 years ago

juefeiyuan commented 6 years ago

Hi, Dr. Chen, I myself have some pairs of sketches and images files, how could I generate the .tfrecord file myself and use your code to run? Thank you

wchen342 commented 6 years ago

The tfrecord files are just a kind of leveldb, and you can find pretty a lot of tutorials on how to build them. The necessary fields are all in data_processing/extract_images.py. I will try to find the script I used when I get back to my computer later this week, if you need it.

juefeiyuan commented 6 years ago

Yes, I need it, I am new to this field, I'll wait for your response, thank you so much.

juefeiyuan commented 6 years ago

Hi, Dr. Chen, Did you find the script to build the .tfrecord files? Thank you for your kindly help.

wchen342 commented 6 years ago

Hi, i created a gist here. It is the one I used to convert Sketchy Database into tfrecords. The csv file used in the script is just used to select which classes I want. You can use any other methods to do the same.

juefeiyuan commented 6 years ago

Hi, Dr. Chen, Thank you so much for your help