Skip to content

Commit 52f7b86

Browse files
committed
Avoid direct changes on the input image in viz
1 parent 07bd24c commit 52f7b86

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
"pyyaml>=5.1",
2727
"torch",
2828
"torchvision",
29-
"pycocotools",
29+
"pycocotools==2.0.1",
3030
"fvcore",
3131
],
3232
extras_require={

src/layoutparser/visualization.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ def wrap(canvas, layout, *args, **kwargs):
104104
if isinstance(canvas, Image.Image):
105105
if canvas.mode != 'RGB':
106106
canvas = canvas.convert('RGB')
107+
canvas = canvas.copy()
107108
elif isinstance(canvas, np.ndarray):
108109
canvas = Image.fromarray(canvas)
109110
out = func(canvas, layout, *args, **kwargs)

0 commit comments

Comments
 (0)