maet3608 / nuts-ml

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

ViewImage and ViewImageAnnotation do not draw when followed by Prefetch/BuildBatch #18

Open maet3608 opened 6 years ago

maet3608 commented 6 years ago

for instance, this does not work (but should) load_images >> ViewImage() >> Prefetch() >> Consume()

The following works: load_images >> Prefetch() >> ViewImage() >> Consume

Similarly the moment a BuildBatch() appears somewhere after ViewImage() drawing does not work because BuildBatch internally uses the same code as Prefetch(). This is threading issue due to the fact that matplotlib is not thread safe.