格物象AI实验——拿铁熊猫实现隐身的斗篷 DF创客社区

#记录下H值

hue_value = mapped_h

hue_range = 20

window_name = "invisible cloak"

print("按下按钮退出...")

while True:

    global if_break

    cap = cv2.VideoCapture(0)

    if cap is not None:

        #print("cap is not none, break")

        break

cv2.namedWindow(window_name, cv2.WND_PROP_FULLSCREEN)

cv2.setWindowProperty(window_name, cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)

if_break = False

whole_frame = np.zeros((480,800,3), dtype="uint8")

while True:

    ret, frame = cap.read()  

    if ret:

        #RGB 三通道

        hsv_mask = get_hsv_mask(frame,hue_value)

        #对掩模取反   

        inv_hsv_mask = cv2.bitwise_not(hsv_mask)

        res = en_invisible(hsv_mask,inv_hsv_mask, frame, background)   

        #res = np.flip(res,axis=1)

        for i in range (3):

            whole_frame[:,80:720,i]=res[:,:,i]

        cv2.imshow(window_name,whole_frame)

    if if_break == True:

        print("退出")

        break

    if cv2.waitKey(1) & 0xFF == ord('q'):

        break

#清空显示

cap.release()

cv2.destroyAllWindows()