Bmo411 commited on
Commit
418696e
·
verified ·
1 Parent(s): eb3df8d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -1
requirements.txt CHANGED
@@ -3,4 +3,23 @@ transformers
3
  gradio
4
  huggingface_hub
5
  numpy
6
- opencv-python
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  gradio
4
  huggingface_hub
5
  numpy
6
+ opencv-python
7
+ # Capa de Transformada de Fourier
8
+ class FourierTransformLayer(Layer):
9
+ def call(self, inputs):
10
+ fourier = tf.signal.fft2d(tf.cast(inputs, tf.complex64))
11
+ fourier = tf.abs(fourier)
12
+ return tf.concat([tf.math.real(fourier), tf.math.imag(fourier)], axis=-1)
13
+
14
+ def compute_output_shape(self, input_shape):
15
+ return (input_shape[0], input_shape[1], input_shape[2], input_shape[3] * 2)
16
+
17
+ # Capa de Transformada Inversa de Fourier
18
+ class InverseFourierTransformLayer(Layer):
19
+ def call(self, inputs):
20
+ real_part, imag_part = tf.split(inputs, num_or_size_splits=2, axis=-1)
21
+ complex_fourier = tf.complex(real_part, imag_part)
22
+ return tf.abs(tf.signal.ifft2d(complex_fourier))
23
+
24
+ def compute_output_shape(self, input_shape):
25
+ return (input_shape[0], input_shape[1], input_shape[2], input_shape[3] // 2)