在跑一篇论文的过程中,为了尽量不改动源码,在PIL端到端的转换过程中、增加了opencv来实现特定的功能,主要是利用opencv库来提取图片显著性的前景。
用到的图片如下:
原图:
mask:
最后的结果:
实现代码:
1、首先是论文里的用PIL库打开rgb图像的函数、返回有两个值、需要用到的是图片转numpy的那个值:pil_to_numpy
import numpy as np
import sys
from PIL import Image
import cv2
np.set_printoptions(threshold=sys.maxsize)
def pil_loader_rgb(path: str) -> Image.Image: