Código de Python - Suavizado de imagen en archivos de vídeo.

Imágen de perfil
Val: 662
Bronce
Ha mantenido su posición en Python (en relación al último mes)
Gráfica de Python
Actualizado

Suavizado de imagen en archivos de vídeo.gráfica de visualizaciones


Python

Actualizado el 9 de Junio del 2021 por Antonio (53 códigos) (Publicado el 24 de Mayo del 2021)
739 visualizaciones desde el 24 de Mayo del 2021
El presente programa se encarga de eliminar el ruido de los fotogramas de un archivo de video empleando la técnica de "filtrado bilateral". El programa realiza el filtrado frame a frame para a continuación generar un nuevo video con la secuencia de frames procesados (aplicando el frame rate del vídeo original). También usa el software "ffmpeg" para copiar el audio del vídeo original y añadirlo al vídeo resultante.

USO: Primeramente seleccionaremos el vídeo a filtrar mediante el botón "SEARCH". Una vez seleccionado iniciaremos el proceso con "START FILTERING" con el que empezaremos seleccionando la ubicación del nuevo vídeo, para a continuación iniciar el proceso (NOTA: La ruta del directorio de destino no deberá contener espacios en blanco). El proceso de filtrado podrá ser cancelado medinate el botón "CANCEL".
PARA CUALQUIER DUDA U OBSERVACIÓN USEN LA SECCIÓN DE COMENTARIOS.

vf

Requerimientos

LENGUAJE: Python
LIBRERÍAS Y RECURSOS: tkinter, os, opencv, ffmpeg, numpy, pydub, mhmovie, threading.

2.1

Actualizado el 9 de Junio del 2021 (Publicado el 24 de Mayo del 2021)gráfica de visualizaciones de la versión: 2.1
485 visualizaciones desde el 24 de Mayo del 2021

2.1.1

Publicado el 9 de Junio del 2021gráfica de visualizaciones de la versión: 2.1.1
254 visualizaciones desde el 9 de Junio del 2021
estrellaestrellaestrellaestrellaestrella
estrellaestrellaestrellaestrella
estrellaestrellaestrella
estrellaestrella
estrella

Esta versión incluye varios métodos de filtrado.

vf19
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
from tkinter import *
from tkinter import ttk
from tkinter import filedialog, messagebox
import cv2 as cv
import ffmpeg
import numpy as np
import threading
from mhmovie.code import *
from pydub import AudioSegment
import tempfile
import os
 
class app:
    def __init__(self):
        self.root = Tk()
        self.root.title("Video Filter")
        self.root.geometry("905x240")#246
        self.root.configure(bg="lavender")
 
        self.currentDir = StringVar()
        self.currentDir.set(os.getcwd())
        self.filename = StringVar()
        self.file = None
        self.canceled = False
        self.frames_list = []
        self.vid_name = None
 
        Entry(self.root,textvariable=self.currentDir,width=158).place(x=0,y=0)
        Entry(self.root,textvariable=self.filename,font=('arial',23,'bold'),width=40).place(x=10,y=25)
        self.btnSearch = Button(self.root,text="SEARCH",height=2,width=25,bg="light blue1",command=self.open_file)
        self.btnSearch.place(x=709,y=25)
        self.btnStart = Button(self.root,text="START FILTERING",width=97,height=2,bg="light green",command=self.init_task)
        self.btnStart.place(x=8,y=77)
        Button(self.root,text="CANCEL",height=2,width=25,bg="light blue1",command=self.cancel).place(x=709,y=77)
        Label(self.root,text="FRAME RATE:",bg="lavender").place(x=710,y=130)
        self.frLabel = Label(self.root,bg='black',width=14,fg="light green")
        self.frLabel.place(x=790,y=130)
        Label(self.root,text="N FRAMES:",bg="lavender").place(x=720,y=170)
        self.nframesLabel = Label(self.root,bg='black',width=14,fg="light green")
        self.nframesLabel.place(x=790,y=170)
        self.prog_bar = ttk.Progressbar(self.root)
        self.prog_bar.place(x=10,y=170,width=687)
        self.processLabel = Label(self.root,text="PROCESS",bg="lavender",width=97)
        self.processLabel.place(x=10,y=148)
        self.filter_method = ttk.Combobox(master=self.root,width=27)
        self.filter_method.place(x=710,y=210)
        self.filter_method["values"]=["Bilateral Filter","Blur","Median Blur","fastNlMeansDenoisingColored"]
        self.filter_method.set("Bilateral Filter")
 
        self.root.mainloop()
 
    def open_file(self):
        self.dir = filedialog.askopenfilename(initialdir="/",title="SELECT FILE",
                        filetypes=(("mp4 files","*.mp4"),("avi files","*.avi"),("gif files","*.gif")))
        if self.dir:
            self.file = self.dir
 
            probe = ffmpeg.probe(self.file)
            self.video_streams = [stream for stream in probe["streams"] if stream["codec_type"] == "video"]
            self.nframes = (self.video_streams[0]['nb_frames'])
            self.height = (self.video_streams[0]['height'])
            self.fr = (self.video_streams[0]['avg_frame_rate'])
            self.size = (self.video_streams[0]['height'])
            self.vidName = (self.file).split("/")[-1]
            self.filename.set(self.vidName)
            self.frLabel.configure(text=self.fr)
            self.nframesLabel.configure(text=self.nframes)
 
    def aply_method(self,fr):
        if self.filter_method.get() == "Bilateral Filter":
            edit = cv.bilateralFilter(fr,9,75,75)
        elif self.filter_method.get() == "Blur":
            edit = cv.blur(fr,(5,5))
        elif self.filter_method.get() == "Median Blur":
            edit = cv.medianBlur(fr,5)
        elif self.filter_method.get() == "fastNlMeansDenoisingColored":
            edit = cv.fastNlMeansDenoisingColored(fr,None,20,10,7,21)
        return edit
 
    def cancel(self):
        self.canceled = True
        self.processLabel.configure(text="CANCELLED")
        self.btnStart.configure(state='normal')
        self.btnSearch.configure(state='normal')
        self.prog_bar.stop()
        #self.counter = 0
        self.frames_list = []
 
    def create_new_video(self):
        frame_array = []
        counter = 0
        dif = 0
        self.question = "yes"
        if len(self.frames_list) > 0:
            for img in self.frames_list:
                if self.canceled == False:
                    counter+=1
                    height, width, layers  = img.shape
                    size = (width,height)
 
                    for k in range(1):
                        frame_array.append(img)
 
                    percent = counter*100/int(self.nframes)
                    self.prog_bar.step(percent-dif)
                    self.processLabel.configure(text="CREATING VIDEO: {}%".format(int(percent)))
                    dif=percent
 
            name,ex = os.path.splitext(self.vidName)
            self.vid_name = (name+'('+self.filter_method.get().replace(" ","")+')'+'.mp4').replace(" ","_")
            if self.vid_name in os.listdir() and self.canceled == False:
                self.question = messagebox.askquestion("OVERWRITE?","{} already exists. Overwrite? [y/N].".format(self.vid_name))
 
            if self.question == "yes" and self.canceled == False:
                if self.vid_name in os.listdir():
                    os.remove(self.vid_name)
                frame_rate = eval(self.fr)
                out = cv.VideoWriter('filteredVideo.mp4',cv.VideoWriter_fourcc(*'mp4v'), frame_rate, size)#'mp4v'
                print("CREATING VIDEO...")
                print('FA:',len(frame_array))
                self.processLabel.configure(text="FINALIZING VIDEO...")
                for i in range(len(frame_array)):
                    out.write(frame_array[i])
 
                out.release()
 
                self.processLabel.configure(text="ADDING AUDIO...")
                if 'VidAudioInfo.mp3' in os.listdir():
                    final_video = movie('filteredVideo.mp4') + music('VidAudioInfo.mp3')
                    print('BOTH')
                else:
                    final_video = movie('filteredVideo.mp4')
                final_video.save(self.vid_name)
 
            self.frames_list = []
 
    def filtering(self):
        if self.file:
            directory = filedialog.askdirectory()
            if directory:
                try:
                    print(os.getcwd())
                    os.chdir(directory)
                    self.btnStart.configure(state='disabled')
                    self.btnSearch.configure(state='disabled')
                    try:
                        self.processLabel.configure(text="GETTING AUDIO DATA...")
                        audio = AudioSegment.from_file(self.file)#
                        audio.export("VidAudioInfo.mp3",format="mp3")
                    except:
                        pass
                    self.currentDir.set(os.getcwd())
                    dif = 0
                    counter = 0
                    self.canceled = False
                    self.cam = cv.VideoCapture(self.file)
                    ret = True
 
                    while self.canceled == False and ret:
                        ret,frame = self.cam.read()
                        if ret:
                            counter+=1
                            name = 'frame'+str(counter)+'.png'
                            edited_frame = self.aply_method(frame)
                            self.frames_list.append(edited_frame)
 
                            percent = counter*100/int(self.nframes)
                            self.prog_bar.step(percent-dif)
                            self.processLabel.configure(text="PROCESSING FRAMES: {} ({}%)".format((counter),int(percent)))
                            dif=percent
 
                    self.create_new_video()
                    self.processLabel.configure(text="PROCESS: ENDED")
                    if self.vid_name and self.canceled == False and self.question == "yes":
                        messagebox.showinfo("TASK COMPLETED","Created video \'{}\'.".format(self.vid_name))
 
                    if 'VidAudioInfo.mp3' in os.listdir():
                        os.remove('VidAudioInfo.mp3')
                    if 'filteredVideo.mp4' in os.listdir():
                        if not self.vid_name in os.listdir():
                            os.rename('filteredVideo.mp4',self.vid_name)
                        else:
                            os.remove('filteredVideo.mp4')
                except Exception as e:
                    messagebox.showwarning("UNEXPECTED ERROR",str(e))
                self.btnStart.configure(state='normal')
                self.btnSearch.configure(state='normal')
 
    def init_task(self):
        t = threading.Thread(target=self.filtering)
        t.start()
 
if __name__=="__main__":
    app()



Comentarios sobre la versión: 2.1.1 (0)


No hay comentarios
 

Comentar la versión: 2.1.1

Nombre
Correo (no se visualiza en la web)
Valoración
Comentarios...
CerrarCerrar
CerrarCerrar
Cerrar

Tienes que ser un usuario registrado para poder insertar imágenes, archivos y/o videos.

Puedes registrarte o validarte desde aquí.

Codigo
Negrita
Subrayado
Tachado
Cursiva
Insertar enlace
Imagen externa
Emoticon
Tabular
Centrar
Titulo
Linea
Disminuir
Aumentar
Vista preliminar
sonreir
dientes
lengua
guiño
enfadado
confundido
llorar
avergonzado
sorprendido
triste
sol
estrella
jarra
camara
taza de cafe
email
beso
bombilla
amor
mal
bien
Es necesario revisar y aceptar las políticas de privacidad

http://lwp-l.com/s7069