import cv2
import numpy as np
import matplotlib.pyplot as plt
i = w = h = f1 = f2 = ew = eh = 0

def cut(event, x, y, flags, param):
  global i,w,h, f1,f2,ew,eh
  img = 0
  #這一個(gè)if的作用是，當(dāng)鼠標(biāo)左鍵按下，記錄開(kāi)始的位置
  if event == cv2.EVENT_LBUTTONDOWN:
    f1 = 1
    h = x
    w = y
    print("w = ", w, "h = ", h)
  #這一步的作用是當(dāng)鼠標(biāo)滑動(dòng)時(shí)，并且左鍵是按下的時(shí)候，就畫(huà)一個(gè)框框，并且記錄左后一個(gè)位置
  if event == cv2.EVENT_MOUSEMOVE and flags == cv2.EVENT_FLAG_LBUTTON:
    img = frame.copy()
    cv2.rectangle(img, (h,w), (x,y), (255,0,0),5)
    cv2.imshow("frame", img)
    f2 = 1
    eh = x
    ew = y
  #當(dāng)執(zhí)行了以上的兩步時(shí)候，并且左鍵是松開(kāi)的，就可以進(jìn)行截出一張圖片。
  if f1 == 1 and f2 == 1 and event == cv2.EVENT_LBUTTONUP:
    #下面兩個(gè)if是向各個(gè)方向上截圖的考慮
    if ew < w:
      w,ew = ew,w
    if eh < h:
      eh,h = h, eh
    i += 1
    f1 = f2 = 0
    img = frame[w:ew, h:eh]
    path = "C:/test2/" + "%d"%i+".jpg"
    cv2.imwrite(path,img)
#這時(shí)創(chuàng)建一個(gè)對(duì)像
cap = cv2.VideoCapture("C:/test2/bbb.mp4")
#這是命名一個(gè)窗口，與下面的鼠標(biāo)結(jié)合，鼠標(biāo)會(huì)在這個(gè)窗口中產(chǎn)生效果
cv2.namedWindow("frame",cv2.WINDOW_NORMAL)
cv2.setMouseCallback("frame",cut)
while True:
 #讀取圖片
  ret, frame = cap.read()
  if ret == False:
    break
  cv2.imshow("frame", frame)
  #如果輸入'c'，就進(jìn)行切割。
  if cv2.waitKey(10) == ord("c"):
    cv2.waitKey(0)
  if cv2.waitKey(10) == ord('q'):
    break