From ee7cc11a72741ae2e197c170e898c4e332fd96d5 Mon Sep 17 00:00:00 2001 From: HaleySchuhl Date: Thu, 31 Oct 2024 16:21:54 -0500 Subject: [PATCH] update prune _debug to plot the other debug img --- plantcv/plantcv/morphology/prune.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/plantcv/plantcv/morphology/prune.py b/plantcv/plantcv/morphology/prune.py index f23644cc9..e71b16dfa 100644 --- a/plantcv/plantcv/morphology/prune.py +++ b/plantcv/plantcv/morphology/prune.py @@ -68,9 +68,6 @@ def prune(skel_img, size=0, mask=None): pruned_img = image_subtract(pruned_img, removed_barbs) pruned_img = _iterative_prune(pruned_img, 1) - # Reset debug mode - params.debug = debug - # Make debugging image if mask is None: pruned_plot = np.zeros(skel_img.shape[:2], np.uint8) @@ -81,10 +78,13 @@ def prune(skel_img, size=0, mask=None): cv2.drawContours(pruned_plot, removed_segments, -1, (0, 0, 255), params.line_thickness, lineType=8) cv2.drawContours(pruned_plot, pruned_obj, -1, (150, 150, 150), params.line_thickness, lineType=8) - _debug(visual=pruned_img, filename=os.path.join(params.debug_outdir, f"{params.device}_pruned.png")) - _debug(visual=pruned_img, filename=os.path.join(params.debug_outdir, f"{params.device}_pruned_debug.png")) - # Segment the pruned skeleton segmented_img, segment_objects = segment_skeleton(pruned_img, mask) + # Reset debug mode + params.debug = debug + + _debug(visual=pruned_img, filename=os.path.join(params.debug_outdir, f"{params.device}_pruned.png")) + _debug(visual=pruned_plot, filename=os.path.join(params.debug_outdir, f"{params.device}_pruned_debug.png")) + return pruned_img, segmented_img, segment_objects