это код я использую:
Код: Выделить всё
import cv2
import tensorflow as tf
import numpy as np
from PyQt5.QtWidgets import (QApplication, QMainWindow, QPushButton, QLabel, QFileDialog, QVBoxLayout,
QWidget, QComboBox, QHBoxLayout, QProgressBar)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QPixmap, QDragEnterEvent, QDropEvent
import os
# Load the pre-trained style transfer model
model = tf.saved_model.load('model/')
def preprocess(image):
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = tf.image.convert_image_dtype(image, tf.float32)
image = tf.image.resize(image, (256, 256))
image = image[tf.newaxis, :]
return image
def apply_style(content_image, style_image):
content_image = preprocess(content_image)
style_image = preprocess(style_image)
stylized_image = model(tf.constant(content_image), tf.constant(style_image))[0]
stylized_image = tf.image.resize(stylized_image, (content_image.shape[1], content_image.shape[2]))
stylized_image = tf.squeeze(stylized_image).numpy()
stylized_image = (stylized_image * 255).astype(np.uint8)
stylized_image = cv2.cvtColor(stylized_image, cv2.COLOR_RGB2BGR)
return stylized_image
class VideoStyleTransferApp(QMainWindow):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
self.setWindowTitle('Video Style Transfer')
self.setGeometry(100, 100, 800, 600)
layout = QVBoxLayout()
self.label = QLabel('Drag and drop a video or click to upload, then select a style', self)
self.label.setAlignment(Qt.AlignCenter)
layout.addWidget(self.label)
self.drop_area = QLabel('Drag and drop video here', self)
self.drop_area.setStyleSheet("QLabel { background-color : lightgray; border: 2px dashed gray; }")
self.drop_area.setAlignment(Qt.AlignCenter)
self.drop_area.setFixedHeight(100)
layout.addWidget(self.drop_area)
self.drop_area.setAcceptDrops(True)
self.drop_area.dragEnterEvent = self.dragEnterEvent
self.drop_area.dropEvent = self.dropEvent
self.upload_video_btn = QPushButton('Upload Video', self)
self.upload_video_btn.clicked.connect(self.upload_video)
layout.addWidget(self.upload_video_btn)
self.style_selector = QComboBox(self)
self.style_selector.addItems(self.load_styles())
layout.addWidget(self.style_selector)
self.process_btn = QPushButton('Process Video', self)
self.process_btn.clicked.connect(self.process_video)
self.process_btn.setEnabled(False)
layout.addWidget(self.process_btn)
self.progress_bar = QProgressBar(self)
self.progress_bar.setValue(0)
self.progress_bar.setTextVisible(True)
layout.addWidget(self.progress_bar)
self.video_path = None
container = QWidget()
container.setLayout(layout)
self.setCentralWidget(container)
def load_styles(self):
self.styles_dir = 'styles/'
styles = [f for f in os.listdir(self.styles_dir) if f.endswith(('.jpg', '.png'))]
return styles
def dragEnterEvent(self, event: QDragEnterEvent):
if event.mimeData().hasUrls():
event.acceptProposedAction()
def dropEvent(self, event: QDropEvent):
for url in event.mimeData().urls():
self.video_path = url.toLocalFile()
if self.video_path.endswith(('.mp4', '.avi')):
self.label.setText(f'Video selected: {self.video_path}')
self.check_files_ready()
def upload_video(self):
options = QFileDialog.Options()
self.video_path, _ = QFileDialog.getOpenFileName(self, 'Open Video File', '', 'Video Files (*.mp4 *.avi)', options=options)
if self.video_path:
self.label.setText(f'Video selected: {self.video_path}')
self.check_files_ready()
def check_files_ready(self):
if self.video_path and self.style_selector.currentText():
self.process_btn.setEnabled(True)
def process_video(self):
if not self.video_path or not self.style_selector.currentText():
return
self.label.setText('Processing video...')
self.progress_bar.setValue(0)
self.repaint()
style_image_path = os.path.join(self.styles_dir, self.style_selector.currentText())
style_image = cv2.imread(style_image_path)
cap = cv2.VideoCapture(self.video_path)
output_folder = os.path.join(os.path.expanduser("~"), "Videos", "StyledVideos")
os.makedirs(output_folder, exist_ok=True)
output_path = os.path.join(output_folder, 'output_' + os.path.basename(self.video_path))
fps = cap.get(cv2.CAP_PROP_FPS)
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
self.progress_bar.setMaximum(frame_count)
processed_frames = 0
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
styled_frame = apply_style(frame, style_image)
out.write(styled_frame)
processed_frames += 1
self.progress_bar.setValue(processed_frames)
cap.release()
out.release()
self.label.setText(f'Processing complete! Output saved as {output_path}')
self.progress_bar.setValue(frame_count) # Ensure progress bar is full at the end
if __name__ == '__main__':
app = QApplication(sys.argv)
ex = VideoStyleTransferApp()
ex.show()
sys.exit(app.exec_())
Результат
вместо этого:
ожиданиеБудем признательны за любое решение, спасибо
Подробнее здесь: https://stackoverflow.com/questions/788 ... -animation
