Empty content images list on specific situation

hwalsuklee / tensorflow-fast-style-transfer

A simple, concise tensorflow implementation of fast style transfer

Apache License 2.0

245 stars 70 forks source link

Empty content images list on specific situation #11

Open Pzoom522 opened 6 years ago

Pzoom522 commented 6 years ago

The content images selection part should be modified. If len(content_images) is divided by batch_size, then self.x_list will be empty, causing failure in training.