opencv cv2.resize 会产生多余的边框 paddle unet 模型输出多出来边框的解决方法

识别出的图片有bug会多出来边框. 解决方法是
修改D:\Anaconda3\Lib\site-packages\paddlex\cv\models\deeplabv3p.py 的第492行,
将

pred = cv2.resize(pred, (w, h), cv2.INTER_NEAREST) 

这一行的代码替换成下面的代码即可

    image = Image.fromarray(pred)
    image = image.resize((w, h),resample=Image.NEAREST) 
    pred = np.array(image) 

别忘记引入

 from PIL import Image, ImageDraw, ImageFont

resize之前的图片

下面是resize之后的图片

用ps就可以看出来,多了一层边框. 貌似还去不掉.

解决方法是用PIL的resize替代opencv的resize

pred = cv2.resize(pred, (w, h))

改成下面的即可

image = Image.fromarray(pred)
image = image.resize((w, h),resample=Image.NEAREST) # resample=Image.NEAREST 参数一定要
# image.save('PIL_resize_pred.png')
pred = np.array(image)