maet3608 / nuts-ml

Flow-based data pre-processing for deep learning
Other
31 stars 10 forks source link

ShowImage does not show image when prefetching is active #1

Open maet3608 opened 7 years ago

maet3608 commented 7 years ago

When building a pipeline such as ShowImage() >> BuildBatch() no images are displayed because BuildBatch performs prefetching. This is due to a threading problem with matplotlib, which is used for image display.

Workaround: Set prefetching to 0 for bathc builder: BuildBatch(prefetch=0)