|
@@ -25,6 +25,6 @@ class SimpleSession(BaseSession):
|
|
|
pred = np.squeeze(pred)
|
|
|
|
|
|
mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
|
|
|
- mask = mask.resize(img.size, Image.Resampling.LANCZOS)
|
|
|
+ mask = mask.resize(img.size, Image.LANCZOS)
|
|
|
|
|
|
return [mask]
|