Image processing with a tensorflow-inspired API - View it on GitHub