fix bug: crop supports gray image

xinntao · Aug 25, 2022 · 3f20f4a · 3f20f4a
1 parent 8dc3cba
commit 3f20f4a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/handyview/utils.py b/handyview/utils.py
@@ -182,6 +182,7 @@ def crop_images(img_list,
 
         # draw rectangle
         if line_width > 0:
-            draw = ImageDraw.Draw(img)
+            img_rect = img.convert('RGB')
+            draw = ImageDraw.Draw(img_rect)
             draw.rectangle(((start_w, start_h), (start_w + len_w, start_h + len_h)), outline=color, width=line_width)
-            img.save(os.path.join(rect_folder, base_name + '_rect.png'))
+            img_rect.save(os.path.join(rect_folder, base_name + '_rect.png'))