diff --git a/app/service/design_fast/pipeline/no_seg_print_painting.py b/app/service/design_fast/pipeline/no_seg_print_painting.py index b6449db..0cf8cf1 100644 --- a/app/service/design_fast/pipeline/no_seg_print_painting.py +++ b/app/service/design_fast/pipeline/no_seg_print_painting.py @@ -169,7 +169,7 @@ class NoSegPrintPainting: canvas_h=painting_dict['dim_image_h'], canvas_w=painting_dict['dim_image_w'], location=painting_dict['location'], - angle=45) + angle=int(print_.get('print_angle_list', [0])[0])) painting_dict['mask_inv_print'] = np.zeros(painting_dict['tile_print'].shape[:2], dtype=np.uint8) return painting_dict diff --git a/app/service/design_fast/pipeline/print_painting.py b/app/service/design_fast/pipeline/print_painting.py index 7b1b2ed..8464762 100644 --- a/app/service/design_fast/pipeline/print_painting.py +++ b/app/service/design_fast/pipeline/print_painting.py @@ -232,7 +232,7 @@ class PrintPainting: canvas_h=painting_dict['dim_image_h'], canvas_w=painting_dict['dim_image_w'], location=painting_dict['location'], - angle=45) + angle=int(print_.get('print_angle_list', [0])[0])) painting_dict['mask_inv_print'] = np.zeros(painting_dict['tile_print'].shape[:2], dtype=np.uint8) return painting_dict