cryptingCompressingVideo/Compression.py at main · YGTRAY/cryptingCompressingVideo · GitHub

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
from matplotlib.image import imread
import numpy as np
import matplotlib.pyplot as plt
import os
import pywt
import threading


def compress_video():
    plt.rcParams['figure.figsize'] = [8, 8]
    plt.rcParams.update({'font.size': 10})


    #input_dir = 'C:\Users\hp\Desktop\.ipynb_checkpoints\Frames'
    #output_dir = 'C:\Users\hp\Desktop\.ipynb_checkpoints\Compressed_Frames'
    input_dir = 'C:/Users/hp/Desktop/pi'
    output_dir = 'C:/Users/hp/Desktop/piCompressed'

    #frame_count = sum([1 for file in os.listdir(input_dir)])


    current_frame=0


    t= os.listdir(input_dir)
    t.sort(key=lambda x: int(x.split(".")[0][5:]))

    for frame in t:
        if frame.endswith('.jpg'):
            # Read image file
            filepath = os.path.join(input_dir, frame)
            print(filepath)

            A = imread(filepath)
            B = np.mean(A, -1); # Convert RGB to grayscale
            ## Wavelet Compression
            n = 5
            w = 'db1'

            coeffs = pywt.wavedec2(B,wavelet=w,level=n)

            coeff_arr, coeff_slices = pywt.coeffs_to_array(coeffs)

            Csort = np.sort(np.abs(coeff_arr.reshape(-1)))

            keep = 0.01
            thresh = Csort[int(np.floor((1-keep)*len(Csort)))]
            ind = np.abs(coeff_arr) > thresh
            Cfilt = coeff_arr * ind # Threshold small indices

            coeffs_filt = pywt.array_to_coeffs(Cfilt,coeff_slices,output_format='wavedec2')

            # Plot reconstruction
            Arecon = pywt.waverec2(coeffs_filt,wavelet=w)

            name = f"{output_dir}/frame{current_frame}.jpg"
            plt.imsave(name, Arecon.astype('uint8'), cmap='gray')


            current_frame += 1


def compress_video_thread():
    """
    This function runs the compress_video() function in a separate thread.
    """
    t = threading.Thread(target=compress_video)
    t.start()

def on_compress_button_click():
    """
    This function is called when the user clicks the compress button.
    """
    compress_video_thread()