Opencv, 给视频加一个进度条 - 图1
© Karobben

由於語法渲染問題而影響閱讀體驗， 請移步博客閱讀～
本文GitPage地址

Opencv Progress Bar

Prepare your video, gif, png

Sample progress bar

We can use a rectangle to be the progress bar

  1. import cv2 as cv2
  3. # Vdieo source
  4. Video = "/run/media/ken/Data/Vlog/Tank_rasbbery/test.avi"
  6. cap=cv2.VideoCapture(Video)
  7. Video_w = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
  8. frame_total = cap.get(cv2.CAP_PROP_FRAME_COUNT)
  9. Num = 0
  10. print(frame_total, Num)
  11. while Num <= frame_total-1:
  12.   # caculate the current frame
  13.   # The width of the progress bar = (current frame /Total frame) * Width of the frame
  14.   Width = int((( Num/frame_total)*420))
  15.   Num +=1
  16.   print(Num, frame_total, width)
  17.   # features for rectangle
  18.   ptLeftTop = (0, 0)
  19.   ptRightBottom = (Width, 20)
  20.   point_color = (0, 0, 255) # BGR
  21.   thickness = 20
  22.   lineType = 8
  23.   # frame
  24.   ret,frame=cap.read()
  25.   frame = cv2.resize(frame, (420,360), interpolation = cv2.INTER_AREA)
  26.   frame = cv2.rectangle(frame, ptLeftTop, ptRightBottom, point_color, thickness, lineType)
  27.   cv2.imshow("video",frame)
  28.   # 在播放每一帧时，使用cv2.waitKey()设置适当的持续时间。如果设置的太低视频就会播放的非常快，如果设置的太高就会播放的很慢。通常情况下25ms就ok
  29.   if cv2.waitKey(25)&0xFF==ord('q'):
  30.      cv2.destroyAllWindows()
  31.      break
  33. print("Mission Down")

loading a picture and moving with progress bar

First, loading the picture

  1. Picture = "/home/ken/Pictures/pokeball.png"
  2. img = cv2.imread(Picture,1)
  3. img = cv2.resize(img, (10,10), interpolation = cv2.INTER_AREA)

Then, picture clean:
Details about threads and mask: 假小牙 2017

Origing by: 万能的小黑Alex 2020
Transparent Background

  1. def CV_mask(img1, img2, rows1, rows2, cols1, cols2):
  2.     # 把logo放在左上角，所以我们只关心这一块区域
  3.     roi = img1[rows1:rows2, cols1:cols2]
  5.     # 创建掩膜
  6.     img2gray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
  7.     print(img2gray[0])
  8.     ret, mask = cv2.threshold(img2gray, 56, 255, cv2.THRESH_BINARY)
  9.     mask_inv = cv2.bitwise_not(mask)
  11.     # 保留除logo外的背景
  12.     img1_bg = cv2.bitwise_and(roi, roi, mask=mask_inv)
  13.     dst = cv2.add(img1_bg, img2) # 进行融合
  14.     img1[rows1:rows2, cols1:cols2] = dst # 融合后放在原图上
  16.     return img1

or

Origing by: 红薯爱帅 2017
Clear Background

  1. def img_deal(img_target, img_logo, row1, row2, col1, col2):
  2.     # 1，对logo进行缩放，按照20%进行
  3.     # cv2.imshow("img_logo", img_logo)
  4.     # 2，对logo做清洗，白色区域是255，其他区域置为黑色0
  5.     img_logo_gray = cv2.cvtColor(img_logo, cv2.COLOR_BGR2GRAY)
  6.     print(img_logo_gray[0])
  7.     ret, img_logo_mask = cv2.threshold(img_logo_gray, 46, 255, cv2.THRESH_BINARY)  # 二值化函数
  8.     img_logo_mask1 = cv2.bitwise_not(img_logo_mask)
  9.     # cv2.imshow("img_logo_gray", img_logo_gray)
  10.     # cv2.imshow("img_logo_mask", img_logo_mask)
  12.     # 3，提取目标图片的ROI
  13.     img_roi = img_target[col1:col2, row1:row2].copy()
  14.     #cv2.imshow("img_roi", img_roi)
  16.     # 4，ROI和Logo图像融合
  17.     img_res0 = cv2.bitwise_and(img_roi, img_roi, mask=img_logo_mask)
  18.     img_res1 = cv2.bitwise_and(img_logo, img_logo, mask=img_logo_mask1)
  19.     img_res2 = cv2.add(img_res0, img_res1)
  20.     # img_res2 = img_res0 + img_res1
  21.     img_target[col1:col2, row1:row2] = img_res2[:, :]
  23.     return img_target

Finally, adding picture to frames

  1. import cv2 as cv2
  2. import numpy as np
  4. # Vdieo source
  5. Video = "/run/media/ken/Data/Vlog/Tank_rasbbery/test.avi"
  6. Picture = "/home/ken/Pictures/16048269.png"
  8. # OUTPUT size of the video
  9. Width_out = int(1920*.9)
  10. Height_out = int(1080*.9)
  11. PB_h_r = 1- 0.5
  12. # Resize image
  13. PB_img_wr = .025
  14. img = cv2.imread(Picture,1)
  15. #img = cv2.resize(img, (200, 300), interpolation = cv2.INTER_AREA)
  17. # progress bar
  18. point_color = (0, 0, 255) # BGR
  19. lineType = 8
  20. thickness_r = 0.005
  22. cap=cv2.VideoCapture(Video)
  23. # config
  24. PB_H = int(Height_out*PB_h_r)
  25. Video_w = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
  26. frame_total = cap.get(cv2.CAP_PROP_FRAME_COUNT)
  27. Num = 0
  28. ## img_position in frame
  29. PB_img_hr = (img.shape[0]/ img.shape[1] )* PB_img_wr
  30. PB_img_w = int(PB_img_wr * Height_out)
  31. PB_img_h = int(PB_img_hr * Height_out)
  32. if PB_img_h%2 == 1:
  33.     PB_img_h += 1
  34. frame_img_h1 = PB_H -int(PB_img_h/2)
  35. frame_img_h2 = PB_H +int(PB_img_h/2)
  36. img = cv2.resize(img, (PB_img_w,PB_img_h), interpolation = cv2.INTER_AREA)
  37. ## Thickness of Progress bar
  38. thickness = int(thickness_r * Height_out)
  40. def img_deal(img_target, img_logo, row1, row2, col1, col2):
  41.     # 1，对logo进行缩放，按照20%进行
  42.     # cv2.imshow("img_logo", img_logo)
  43.     # 2，对logo做清洗，白色区域是255，其他区域置为黑色0
  44.     img_logo_gray = cv2.cvtColor(img_logo, cv2.COLOR_BGR2GRAY)
  45.     print(img_logo_gray[0])
  46.     ret, img_logo_mask = cv2.threshold(img_logo_gray, 46, 255, cv2.THRESH_BINARY)  # 二值化函数
  47.     img_logo_mask1 = cv2.bitwise_not(img_logo_mask)
  48.     # cv2.imshow("img_logo_gray", img_logo_gray)
  49.     # cv2.imshow("img_logo_mask", img_logo_mask)
  51.     # 3，提取目标图片的ROI
  52.     img_roi = img_target[col1:col2, row1:row2].copy()
  53.     #cv2.imshow("img_roi", img_roi)
  55.     # 4，ROI和Logo图像融合
  56.     img_res0 = cv2.bitwise_and(img_roi, img_roi, mask=img_logo_mask)
  57.     img_res1 = cv2.bitwise_and(img_logo, img_logo, mask=img_logo_mask1)
  58.     img_res2 = cv2.add(img_res0, img_res1)
  59.     # img_res2 = img_res0 + img_res1
  60.     img_target[col1:col2, row1:row2] = img_res2[:, :]
  62.     return img_target
  64. def CV_mask(img1, img2, rows1, rows2, cols1, cols2):
  65.     # 把logo放在左上角，所以我们只关心这一块区域
  66.     roi = img1[rows1:rows2, cols1:cols2]
  68.     # 创建掩膜
  69.     img2gray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
  70.     print(img2gray[0])
  71.     ret, mask = cv2.threshold(img2gray, 46, 255, cv2.THRESH_BINARY_INV)
  72.     mask_inv = cv2.bitwise_not(mask)
  74.     # 保留除logo外的背景
  75.     img1_bg = cv2.bitwise_and(roi, roi, mask=mask_inv)
  76.     dst = cv2.add(img1_bg, img2) # 进行融合
  77.     img1[rows1:rows2, cols1:cols2] = dst # 融合后放在原图上
  79.     return img1
  81. while Num <= frame_total-1:
  82.   # caculate the current frame
  83.   # The width of the progress bar = (current frame /Total frame) * Width of the frame
  84.   Width = int((( Num/frame_total)*Width_out))
  85.   Num +=1
  86.   # features for rectangle
  87.   ptLeftTop = (0, PB_H)
  88.   print(Num, frame_total, Width, PB_H, print(img[0][0]))
  89.   ptRightBottom = (Width, PB_H)
  90.   # frame
  91.   ret,frame=cap.read()
  92.   frame = cv2.resize(frame, (Width_out,Height_out), interpolation = cv2.INTER_AREA)
  93.   frame = cv2.rectangle(frame, ptLeftTop, ptRightBottom, point_color, thickness, lineType)
  94.   # Adding png for progress bar
  95.   Width_png = int((( Num/frame_total)*(Width_out-PB_img_w)))
  96.   # 保留除logo外的背景
  97.   #frame =
  98.   CV_mask(frame, img, frame_img_h1, frame_img_h2, 0+Width_png, PB_img_w+Width_png)
  99.   #frame = img_deal(frame, img, 0+Width_png, PB_img_w+Width_png, frame_img_h1, frame_img_h2)
  100.   #frame[frame_img_h1:frame_img_h2 ,0+Width_png:PB_img_w+Width_png] = img
  101.   # frame show
  102.   cv2.imshow("video",frame)
  103.   # 在播放每一帧时，使用cv2.waitKey()设置适当的持续时间。如果设置的太低视频就会播放的非常快，如果设置的太高就会播放的很慢。通常情况下25ms就ok
  104.   if cv2.waitKey(25)&0xFF==ord('q'):
  105.      cv2.destroyAllWindows()
  106.      break
  108. print("Mission Down")

Gif progress

Loading gif

  1. import time, cv2
  2. from PIL import Image
  3. import numpy as np
  5. file = "/home/ken/Pictures/pikachu_run.gif"
  7. List = []
  8. im = Image.open(file)
  9. im.seek(1)#skip to the second frame
  10. try:
  11.     while 1:
  12.         List += [cv2.cvtColor(np.asarray(im.convert()),cv2.COLOR_RGB2BGR)]
  13.         im.seek(im.tell()+1)
  14. except EOFError:#the sequence ends
  15.     pass
  16. Num = 0
  17. while Num < len(List)*2:
  18.     Num +=1
  19.     ID = Num%(len(List))
  20.     cv2.imshow("OpenCV",List[ID])
  21.     print(ID)
  22.     cv2.waitKey(1)
  23.     time.sleep(0.1)
  24. cv2.destroyAllWindows()

All codes

  1. import cv2 as cv2
  2. import numpy as np
  3. from PIL import Image
  5. # Vdieo source
  6. Video = "/run/media/ken/Data/Vlog/Tank_rasbbery/test.avi"
  7. Picture = "/home/ken/Pictures/test.gif"
  9. # OUTPUT size of the video
  10. Width_out = int(1920*.9)
  11. Height_out = int(1080*.9)
  12. PB_h_r = 1- 0.05
  13. # Resize image
  14. PB_img_wr = .1
  15. img = cv2.imread(Picture,1)
  16. #img = cv2.resize(img, (200, 300), interpolation = cv2.INTER_AREA)
  18. # progress bar
  19. point_color = (0, 0, 255) # BGR
  20. lineType = 8
  21. thickness_r = 0.005
  23. cap=cv2.VideoCapture(Video)
  24. # config
  25. PB_H = int(Height_out*PB_h_r)
  26. Video_w = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
  27. frame_total = cap.get(cv2.CAP_PROP_FRAME_COUNT)
  28. Num = 0
  29. ## img_position in frame
  30. #PB_img_hr = (img.shape[0]/ img.shape[1] )* PB_img_wr
  31. #PB_img_w = int(PB_img_wr * Height_out)
  32. #PB_img_h = int(PB_img_hr * Height_out)
  33. #if PB_img_h%2 == 1:
  34. #    PB_img_h += 1
  35. #frame_img_h1 = PB_H -int(PB_img_h/2)
  36. #frame_img_h2 = PB_H +int(PB_img_h/2)
  37. #img = cv2.resize(img, (PB_img_w,PB_img_h), interpolation = cv2.INTER_AREA)
  38. ## Thickness of Progress bar
  39. thickness = int(thickness_r * Height_out)
  41. def img_deal(img_target, img_logo, row1, row2, col1, col2, THRE):
  42.     # 1，对logo进行缩放，按照20%进行
  43.     # cv2.imshow("img_logo", img_logo)
  44.     # 2，对logo做清洗，白色区域是255，其他区域置为黑色0
  45.     img_logo_gray = cv2.cvtColor(img_logo, cv2.COLOR_BGR2GRAY)
  46.     print(img_logo_gray[0])
  47.     ret, img_logo_mask = cv2.threshold(img_logo_gray, THRE, 255, cv2.THRESH_BINARY)  # 二值化函数
  48.     img_logo_mask1 = cv2.bitwise_not(img_logo_mask)
  49.     # cv2.imshow("img_logo_gray", img_logo_gray)
  50.     # cv2.imshow("img_logo_mask", img_logo_mask)
  52.     # 3，提取目标图片的ROI
  53.     img_roi = img_target[col1:col2, row1:row2].copy()
  54.     #cv2.imshow("img_roi", img_roi)
  56.     # 4，ROI和Logo图像融合
  57.     img_res0 = cv2.bitwise_and(img_roi, img_roi, mask=img_logo_mask)
  58.     img_res1 = cv2.bitwise_and(img_logo, img_logo, mask=img_logo_mask1)
  59.     img_res2 = cv2.add(img_res0, img_res1)
  60.     # img_res2 = img_res0 + img_res1
  61.     img_target[col1:col2, row1:row2] = img_res2[:, :]
  63.     return img_target
  65. # reading GIF file
  66. List = []
  67. im = Image.open(Picture)
  68. im.seek(1)#skip to the second frame
  69. try:
  70.     while 1:
  71.         img = cv2.cvtColor(np.asarray(im.convert()),cv2.COLOR_RGB2BGR)
  72.         PB_img_hr = (img.shape[0]/ img.shape[1] )* PB_img_wr
  73.         PB_img_w = int(PB_img_wr * Height_out)
  74.         PB_img_h = int(PB_img_hr * Height_out)
  75.         if PB_img_h%2 == 1:
  76.             PB_img_h += 1
  77.         frame_img_h1 = PB_H -int(PB_img_h/2)
  78.         frame_img_h2 = PB_H +int(PB_img_h/2)
  80.         List += [cv2.resize(img, (PB_img_w,PB_img_h), interpolation = cv2.INTER_AREA)]
  81.         im.seek(im.tell()+1)
  82. except EOFError:#the sequence ends
  83.     pass
  85. def CV_mask(img1, img2, rows1, rows2, cols1, cols2, THRE):
  86.     # 把logo放在左上角，所以我们只关心这一块区域
  87.     roi = img1[rows1:rows2, cols1:cols2]
  89.     # 创建掩膜
  90.     img2gray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
  91.     print(img2gray[0])
  92.     ret, mask = cv2.threshold(img2gray, THRE, 255, cv2.THRESH_BINARY_INV)
  93.     mask_inv = cv2.bitwise_not(mask)
  95.     # 保留除logo外的背景
  96.     img1_bg = cv2.bitwise_and(roi, roi, mask=mask_inv)
  97.     dst = cv2.add(img1_bg, img2) # 进行融合
  98.     img1[rows1:rows2, cols1:cols2] = dst # 融合后放在原图上
  100.     return img1
  102. while Num <= frame_total-1:
  103.   # caculate the current frame
  104.   # The width of the progress bar = (current frame /Total frame) * Width of the frame
  105.   Width = int((( Num/frame_total)*(Width_out-PB_img_w))) + int(PB_img_w*.5)
  106.   Num +=1
  107.   # features for rectangle
  108.   ptLeftTop = (0, PB_H)
  109.   img = List[int(Num%len(List))]
  110.   img = cv2.flip(img,1 )
  111.   print(Num, frame_total, Width, PB_H, print(img[0][0]))
  112.   ptRightBottom = (Width, PB_H)
  113.   # frame
  114.   ret,frame=cap.read()
  115.   frame = cv2.resize(frame, (Width_out,Height_out), interpolation = cv2.INTER_AREA)
  116.   frame = cv2.rectangle(frame, ptLeftTop, ptRightBottom, point_color, thickness, lineType)
  117.   # Adding png for progress bar
  118.   Width_png = int((( Num/frame_total)*(Width_out-PB_img_w)))
  119.   # 保留除logo外的背景
  120.   #frame =
  121.   #CV_mask(frame, img, frame_img_h1, frame_img_h2, 0+Width_png, PB_img_w+Width_png, 207)
  122.   frame = img_deal(frame, img, 0+Width_png, PB_img_w+Width_png, frame_img_h1, frame_img_h2, 207)
  123.   #frame[frame_img_h1:frame_img_h2 ,0+Width_png:PB_img_w+Width_png] = img
  124.   # frame show
  125.   cv2.imshow("video",frame)
  126.   # 在播放每一帧时，使用cv2.waitKey()设置适当的持续时间。如果设置的太低视频就会播放的非常快，如果设置的太高就会播放的很慢。通常情况下25ms就ok
  127.   if cv2.waitKey(25)&0xFF==ord('q'):
  128.      cv2.destroyAllWindows()
  129.      break
  131. print("Mission Down")

Enjoy~

本文由Python腳本GitHub/語雀自動更新

由於語法渲染問題而影響閱讀體驗， 請移步博客閱讀～
本文GitPage地址

GitHub: Karobben
Blog:Karobben
BiliBili:史上最不正經的生物狗