aboutsummaryrefslogtreecommitdiffstats
path: root/libavfilter/dnn_backend_tf.c
diff options
context:
space:
mode:
authorSergey Lavrushkin <dualfal@gmail.com>2018-07-28 12:55:02 +0300
committerPedro Arthur <bygrandao@gmail.com>2018-08-07 13:28:04 -0300
commita66e74306a36c8467452a67d2d1af51dbbb9b4c5 (patch)
treeeed70c469f220b890eaef19f14a7d305d523bcf1 /libavfilter/dnn_backend_tf.c
parent9d87897ba84a3b639a4c3afeb4ec6d21bc306a92 (diff)
downloadffmpeg-a66e74306a36c8467452a67d2d1af51dbbb9b4c5.tar.gz
libavfilter: Fixes warnings for unused variables in dnn_srcnn.h, dnn_espcn.h, dnn_backend_tf.c.
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn_backend_tf.c')
-rw-r--r--libavfilter/dnn_backend_tf.c66
1 files changed, 64 insertions, 2 deletions
diff --git a/libavfilter/dnn_backend_tf.c b/libavfilter/dnn_backend_tf.c
index 6528a2a390..bd21137a8a 100644
--- a/libavfilter/dnn_backend_tf.c
+++ b/libavfilter/dnn_backend_tf.c
@@ -375,9 +375,71 @@ DNNModel *ff_dnn_load_default_model_tf(DNNDefaultModel model_type)
TFModel *tf_model = NULL;
TF_OperationDescription *op_desc;
TF_Operation *op;
- TF_Operation *const_ops_buffer[6];
TF_Output input;
- int64_t input_shape[] = {1, -1, -1, 1};
+ static const int64_t input_shape[] = {1, -1, -1, 1};
+ static const char tanh[] = "Tanh";
+ static const char sigmoid[] = "Sigmoid";
+ static const char relu[] = "Relu";
+
+ static const float *srcnn_consts[] = {
+ srcnn_conv1_kernel,
+ srcnn_conv1_bias,
+ srcnn_conv2_kernel,
+ srcnn_conv2_bias,
+ srcnn_conv3_kernel,
+ srcnn_conv3_bias
+ };
+ static const long int *srcnn_consts_dims[] = {
+ srcnn_conv1_kernel_dims,
+ srcnn_conv1_bias_dims,
+ srcnn_conv2_kernel_dims,
+ srcnn_conv2_bias_dims,
+ srcnn_conv3_kernel_dims,
+ srcnn_conv3_bias_dims
+ };
+ static const int srcnn_consts_dims_len[] = {
+ 4,
+ 1,
+ 4,
+ 1,
+ 4,
+ 1
+ };
+ static const char *srcnn_activations[] = {
+ relu,
+ relu,
+ relu
+ };
+
+ static const float *espcn_consts[] = {
+ espcn_conv1_kernel,
+ espcn_conv1_bias,
+ espcn_conv2_kernel,
+ espcn_conv2_bias,
+ espcn_conv3_kernel,
+ espcn_conv3_bias
+ };
+ static const long int *espcn_consts_dims[] = {
+ espcn_conv1_kernel_dims,
+ espcn_conv1_bias_dims,
+ espcn_conv2_kernel_dims,
+ espcn_conv2_bias_dims,
+ espcn_conv3_kernel_dims,
+ espcn_conv3_bias_dims
+ };
+ static const int espcn_consts_dims_len[] = {
+ 4,
+ 1,
+ 4,
+ 1,
+ 4,
+ 1
+ };
+ static const char *espcn_activations[] = {
+ tanh,
+ tanh,
+ sigmoid
+ };
input.index = 0;