OpenCV-Python/middle102.py at master · S2ej1n/OpenCV-Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import numpy as np, cv2

background_img = cv2.imread("background.jpg",cv2.IMREAD_COLOR)
logo = cv2.imread("logo.jpg",cv2.IMREAD_COLOR)

title = "mid102"
cv2.namedWindow(title)

cv2.imshow(title,background_img)

def red_bar(value):
    pass
def green_bar(value):
    pass
def blue_bar(value):
    pass

blue, green, red = cv2.split(background_img)

font = cv2.FONT_HERSHEY_PLAIN

def onMouse(event, x, y, flags, g):
    count = 0

    if event == cv2.EVENT_LBUTTONDOWN:
        print("왼쪽")
        #원래 이미지에 로고 합성
        mask = cv2.threshold(logo, 220, 255, cv2.THRESH_BINARY)[1]
        masks = cv2.split(mask)
        fg_mask = cv2.bitwise_or(masks[0], masks[1])
        fg_mask = cv2.bitwise_or(masks[2], fg_mask)
        bg_mask = cv2.bitwise_not(fg_mask)

        (H, W), (h, w) = background_img.shape[:2], logo.shape[:2]
        x, y = (H - h) // 2, (W - w) // 2
        roi = background_img[x:x + h, y:y + w]

        newlogo = cv2.bitwise_and(logo, logo, mask=fg_mask)
        newback = cv2.bitwise_and(roi, roi, mask=bg_mask)
        dst = cv2.add(newlogo, newback)
        background_img[x:x + h, y:y + w] = dst

        cv2.imshow(title, background_img)
    elif event == cv2.EVENT_RBUTTONDOWN:
        print("오른쪽")
        cv2.rectangle(background_img,(x,y),(x+30,y+30),(255,0,0),2)
        count = count + 1
        cv2.putText(background_img, str(count), (10 + count, 330), font, 2, (255, 0, 0))
        cv2.imshow(title, background_img)

cv2.createTrackbar("red",title,0,100,red_bar)
cv2.createTrackbar("green",title,0,255,green_bar)
cv2.createTrackbar("blue",title,0,255,blue_bar)

while True:
    cv2.setMouseCallback(title, onMouse)
    if cv2.waitKey(20) == 27:
        break
    r = cv2.getTrackbarPos("red",title)
    g = cv2.getTrackbarPos("green", title)
    b = cv2.getTrackbarPos("blue", title)

    cv2.add(red,r,red)
    cv2.add(green,g,green)
    cv2.add(blue,b,blue)

    background_img = cv2.merge([blue,green,red])
    # cv2.imshow(title,background_img)

cv2.imshow(title,background_img)
cv2.setMouseCallback(title,onMouse)
cv2.waitKey()
cv2.destroyAllWindows()