Skip to content

Commit f0d6567

Browse files
committed
Removed PIL warning
the "mode" option is deprecated by new Pillow module
1 parent 84ba324 commit f0d6567

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

slvideotools/datagen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def __init__(self, dest_dir, base_frame_name: str = "frame-", img_ext: str = "pn
190190
def consume(self, frame: np.ndarray):
191191

192192
# Convert into PIL Image
193-
img: Image = PIL.Image.fromarray(frame, 'RGB')
193+
img: Image = PIL.Image.fromarray(frame)
194194

195195
# Compose name and seve
196196
img_filename = self._frame_name + "{:06d}.{}".format(self._img_counter, self._img_ext)

slvideotools/draw_bbox.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def draw_bbox(frames_in: FrameProducer, bbox: Tuple[int, int, int, int], frames_
1818

1919
for frame in frames_in.frames():
2020

21-
frame_pil: Image = PIL.Image.fromarray(obj=frame, mode='RGB')
21+
frame_pil: Image = PIL.Image.fromarray(obj=frame)
2222

2323
draw = ImageDraw.Draw(frame_pil)
2424
draw.rectangle(xy=[bbox[0], bbox[1], bbox[0] + bbox[2], bbox[1] + bbox[3]], outline=(220, 10, 10))

0 commit comments

Comments
 (0)