search

wchen342 / SketchyGAN

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

flickr filter #13

Closed YellowPig-zp closed 5 years ago

YellowPig-zp commented 5 years ago

I am using the provided flickr_crawler.py to retrieve images. For each category ~100,000 images are retrieved. But flickr_filter.py sometimes filters out all the images in a category. Is there any workaround to this? Thanks!

wchen342 commented 5 years ago

What do you mean by 'filters out all the images in a category'? A little more details will help me understand what was going on. Did you use the function for ImageNet categories to filter MSCOCO categories, or vice versa? That is one of reasons I can think of.

YellowPig-zp commented 5 years ago

Hi Dr.Chen. Thanks for the reply. I am using inception_v2, as suggested by the code, to filter the 31 classes in imagenet_share_classes.txt. The program removes all the images in some classes but works fine for other classes.

wchen342 commented 5 years ago

This is strange. You can try printing out predictions and probabilities the model outputs for images in those faulty classes. If the predictions are correct then the problem is in the python code, otherwise there is something wrong with the model.