feat(api): save output of each stage in debug mode

ssube · Jan 29, 2023 · 680adc7 · 680adc7
1 parent 8f1cbc8
commit 680adc7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py
@@ -96,6 +96,9 @@ def stage_tile(tile: Image.Image, _dims) -> Image.Image:
             logger.info('finished stage %s, result size: %sx%s',
                         name, image.width, image.height)
 
+            if is_debug():
+                image.save(path.join(ctx.output_path, 'last-stage.png'))
+
         end = monotonic()
         duration = timedelta(seconds=(end - start))
         logger.info('finished pipeline in %s, result size: %sx%s',