from keras.preprocessing import image import glob import cv2 #step0:给定背景图集的路径,目标物的路径,以及换完背景的目标物保存路径 GroPath = '/home/zhen/DATASET/验证放射变换可以提取object/背景图片/' SouPath = '/home/zhen/DATASET/验证放射变换可以提取object/背景图片/source/caomei200*200.jpg' ResPath = '/home/zhen/DATASET/验证放射变换可以提取object/背景图片换完的目标物/' #step1:换背景 count = 1 imS = image.load_img(SouPath) #循环换的背景图片 for imgGro in glob.glob(GroPath+'*.jpg'): imG = image.load_img(imgGro) imGT = image.load_img(imgGro) #将目标物放在背景图片的中间 for i in range(200): for j in range(200): imGT.putpixel((100 + i, 100 + j), imS.getpixel((i, j))) #将黑色部分用背景原来坐标点的像素填充 for i in range(400): for j in range(400): r, b, g = imGT.getpixel((i, j)) if r + b + g < 10: imGT.putpixel((i, j), imG.getpixel((i, j))) im = image.img_to_array(imGT) im = cv2.cvtColor(im,cv2.COLOR_RGB2BGR) cv2.imwrite(ResPath+str(count)+'.jpg',im) count += 1
keras中给目标物换背景
最新推荐文章于 2021-11-15 01:49:55 发布