merge.py

import numpy as np
import cv2
import time
import smtplib
import os
import sys
from matplotlib import pyplot as plt

#check image database
dataImage = ["bg_1.jpg","bg_2.jpg","bg_3.jpg","bg_4.jpg"]
pathImg = sys.argv[1]
if pathImg in dataImage :
    index = dataImage.index(pathImg)
# Load two images
img1 = cv2.imread(str(pathImg))
img2 = cv2.imread('img_transparent.png')

# I want to put logo on top-left corner, So I create a ROI
rows,cols,channels = img2.shape
roi = img1[0:rows, 0:cols ]

# Now create a mask of logo and create its inverse mask also
img2gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
ret, mask = cv2.threshold(img2gray, 10, 255, cv2.THRESH_BINARY)
mask_inv = cv2.bitwise_not(mask)

# Now black-out the area of logo in ROI
img1_bg = cv2.bitwise_and(roi,roi,mask = mask_inv)

# Take only region of logo from logo imagmergee.
img2_fg = cv2.bitwise_and(img2,img2,mask = mask)

# Put logo in ROI and modify the main image
dst = cv2.add(img1_bg,img2_fg)
img1[0:rows, 0:cols ] = dst

cv2.imwrite('img_result'+str(index)+'.png',img1)

print ('close')
sys.stdout.flush()