【整活】python实现肌肉金轮词云图视频，堪称中了金轮的毒

import cv2
cap = cv2.VideoCapture(r"output.mp4")num = 1while True:    # 逐帧读取视频  按顺序保存到本地文件夹    ret, frame = cap.read()    if ret:        if 1 <= num < 1700:            cv2.imwrite(f"./test_data/test_images/img_{num}.jpg", frame)   # 保存一帧帧的图片            print(f'========== 已成功保存第{num}张图片 ==========')        num += 1    else:        breakcap.release()   # 释放资源

Python 3.6numpy 1.15.2scikit-image 0.14.0PIL 5.2.0PyTorch 0.4.0torchvision 0.2.1glob

git clone https://github.com/NathanUA/BASNet.git

python basnet_test.py

# -*- coding: UTF-8 -*-
from wordcloud import WordCloudimport collectionsimport jiebaimport refrom PIL import Imageimport matplotlib.pyplot as pltimport numpy as np

# 读取数据with open('jinlun.txt') as f:    data = f.read()
# 文本预处理  去除一些无用的字符   只提取出中文出来new_data = re.findall('[\u4e00-\u9fa5]+', data, re.S)new_data = "/".join(new_data)
# 文本分词seg_list_exact = jieba.cut(new_data, cut_all=True)
result_list = []with open('stop_words.txt', encoding='utf-8') as f:    con = f.read().split('\n')    stop_words = set()    for i in con:        stop_words.add(i)
for word in seg_list_exact:    # 设置停用词并去除单个词    if word not in stop_words and len(word) > 1:        result_list.append(word)
# 筛选后统计词频word_counts = collections.Counter(result_list)path = './wordcloud/'
for num in range(1, 1668):    img = f'./test_data/test_results/img_{num}.png'    # 获取蒙版图片    mask_ = 255 - np.array(Image.open(img))    # 绘制词云    plt.figure(figsize=(8, 5), dpi=200)    my_cloud = WordCloud(        background_color='black',  # 设置背景颜色  默认是black        mask=mask_,      # 自定义蒙版        mode='RGBA',        max_words=500,        font_path='simhei.ttf',   # 设置字体  显示中文    ).generate_from_frequencies(word_counts)
    # 显示生成的词云图片    plt.imshow(my_cloud)    # 显示设置词云图中无坐标轴    plt.axis('off')    word_cloud_name = path + 'wordcloud_{}.png'.format(num)    my_cloud.to_file(word_cloud_name)    # 保存词云图片    print(f'======== 第{num}张词云图生成 ========')

import cv2
for i in range(1,1668):    src1_path = "./test_data/test_images/img_{}.jpg".format(i)    src1 = cv2.imread(src1_path)    src2_path = "./wordcloud/wordcloud_{}.png".format(i)    src2 = cv2.imread(src2_path)
    dst = cv2.addWeighted(src1, 1, src2, 0.5,0)    dst_img = "./result/{}.png".format(i)    cv2.imwrite(dst_img,dst)    print("img process {} !".format(dst_img))

# 输出视频的保存路径video_dir = 'result-1.mp4'# 帧率fps = 24# 图片尺寸test = cv2.imread("./test_data/test_images/img_1.jpg")x,y,_ = test.shape img_size = (y, x)
fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')  # opencv3.0 mp4会有警告但可以播放videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
for i in range(1, 1668):    img_path = './result/' + '{}.png'.format(i)    frame = cv2.imread(img_path)    frame = cv2.resize(frame, img_size)   # 生成视频   图片尺寸和设定尺寸相同    videoWriter.write(frame)      # 写进视频里    print(f'======== 按照视频顺序第{i}张图片合进视频 ========')
videoWriter.release()   # 释放资源

import moviepy.editor as mpy
# 读取词云视频my_clip = mpy.VideoFileClip('result-1.mp4')# 截取背景音乐audio_background = mpy.AudioFileClip('output.mp4')audio_background.write_audiofile('output.mp3')# 视频中插入音频final_clip = my_clip.set_audio(audio_background)# 保存为最终的视频   动听的音乐！漂亮小姐姐词云跳舞视频！final_clip.write_videofile('final_video.mp4')