MCPcopy Create free account

hub / github.com/LucasSheng/avatar-net / functions

Functions91 in github.com/LucasSheng/avatar-net

↓ 6 callersFunctionconv2d_same
strided 2-D convolution with 'REFLECT' padding. Args: inputs: A 4-D tensor of size [batch, height, width, channel] num_outputs: A
models/network_ops.py:12
↓ 3 callersFunction_aspect_preserving_resize
(image, smallest_side)
models/preprocessing.py:150
↓ 3 callersFunction_convert_dataset
(split_name, image_filenames, dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:65
↓ 3 callersFunctionbytes_feature
(values)
datasets/dataset_utils.py:28
↓ 2 callersFunction_crop
(image, offset_height, offset_width, crop_height, crop_width)
models/preprocessing.py:19
↓ 2 callersFunction_get_dataset_filename
(dataset_dir, split_name, shard_id)
datasets/convert_mscoco_to_tfexamples.py:59
↓ 2 callersFunction_mean_image_subtraction
(image, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:122
↓ 2 callersFunctionconv2d_resize
deconvolution alternatively with the conv2d_transpose, where we first resize the inputs, and then convolve the results, see http://distill
models/network_ops.py:43
↓ 2 callersMethoddecode_image
(self, sess, image_data)
datasets/dataset_utils.py:137
↓ 2 callersFunctionget_image_filenames
helper fn that provides the full image filenames from the dataset_dir
evaluate_style_transfer.py:44
↓ 2 callersFunctionimage_reader
help fn that provides numpy image coding utilities
evaluate_style_transfer.py:53
↓ 2 callersFunctionproject_features
(features, projection_module='ZCA')
models/avatar_net.py:278
↓ 1 callersFunction_central_crop
(image_list, crop_height, crop_width)
models/preprocessing.py:108
↓ 1 callersFunction_configure_learning_rate
Configures the learning rate. Args: num_samples_per_epoch: The number of samples in each epoch of training global_step: The globa
train_image_reconstruction.py:138
↓ 1 callersFunction_configure_optimizer
Configures the optimizer used for training. Args: learning_rate: A scalar or 'Tensor' learning rate Returns: An instance of
train_image_reconstruction.py:177
↓ 1 callersFunction_dataset_exists
(dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:95
↓ 1 callersFunction_get_filenames
(dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:28
↓ 1 callersFunction_get_init_fn
Returns a function to warm-start the training. Note that the init_fn is only run when initializing the models during the very first global st
train_image_reconstruction.py:245
↓ 1 callersFunction_get_variables_to_train
Returns a list of variables to train. Args: A list of variables to train by the optimizer.
train_image_reconstruction.py:227
↓ 1 callersFunction_random_crop
(image_list, crop_height, crop_width)
models/preprocessing.py:45
↓ 1 callersFunction_smallest_size_at_least
(height, width, smallest_side)
models/preprocessing.py:135
↓ 1 callersFunctionadain_colorization
(normalized_features, colorization_kernels, mean_features)
models/avatar_net.py:424
↓ 1 callersFunctionadain_normalization
(features)
models/avatar_net.py:416
↓ 1 callersMethodauto_encoder
(self, inputs, content_layer=2, reuse=True)
models/autoencoder.py:46
↓ 1 callersFunctionbatch_mean_image_subtraction
(images, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:210
↓ 1 callersMethodbuild_train_graph
build the training graph for the training of the hierarchical autoencoder
models/avatar_net.py:158
↓ 1 callersFunctioncompute_content_features
compute the content features from the end_point dict
models/losses.py:42
↓ 1 callersFunctioncompute_gram_matrix
compute the gram matrix for a layer of feature the gram matrix is normalized with respect to the samples and the dimensions of the input
models/losses.py:17
↓ 1 callersFunctioncompute_style_features
compute the style features from the end_point dict
models/losses.py:52
↓ 1 callersFunctioncompute_sufficient_statistics
compute the gram matrix for a layer of feature
models/losses.py:34
↓ 1 callersMethoddecode_image
(self, sess, image_data)
datasets/dataset_utils.py:120
↓ 1 callersFunctionextract_image_features
compute the dict of layer-wise image features from a given list of networks Args: inputs: the inputs image should be normalized between [-1
models/losses.py:72
↓ 1 callersMethodget_training_operations
(self, optimizer, global_step,
models/avatar_net.py:211
↓ 1 callersFunctionhas_dataset_meta_data_file
(dataset_dir, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:49
↓ 1 callersMethodhierarchical_autoencoder
hierarchical autoencoder for content reconstruction
models/avatar_net.py:135
↓ 1 callersFunctionimsave
(filename, img)
evaluate_style_transfer.py:63
↓ 1 callersFunctioninstance_norm
(inputs, epsilon=1e-10)
models/network_ops.py:90
↓ 1 callersFunctionint64_feature
(values)
datasets/dataset_utils.py:22
↓ 1 callersFunctionmean_image_subtraction
works for one single image with dynamic shapes
models/preprocessing.py:192
↓ 1 callersFunctionnearest_patch_swapping
(content_features, style_features, patch_size=3)
models/avatar_net.py:299
↓ 1 callersFunctionpreprocessing_for_eval
(image, output_height, output_width, resize_side)
models/preprocessing.py:173
↓ 1 callersFunctionpreprocessing_for_train
(image, output_height, output_width, resize_side)
models/preprocessing.py:165
↓ 1 callersFunctionpreprocessing_image
(image, output_height, output_width, resize_side=_RESIZE_SIDE_MIN, is_training=False)
models/preprocessing.py:181
↓ 1 callersFunctionread_dataset_meta_data
(dataset_dir, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:53
↓ 1 callersMethodread_image_dims
(self, sess, image_data)
datasets/dataset_utils.py:116
↓ 1 callersFunctionreconstruct_features
(projected_features, feature_kernels, mean_features,
models/avatar_net.py:287
↓ 1 callersFunctionrun
(input_dataset_dir, output_dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:105
↓ 1 callersFunctionstyle_decorator
style decorator for high-level feature interaction Args: content_features: a tensor of size [batch_size, height, width, channel]
models/avatar_net.py:235
↓ 1 callersFunctionsubsample
(inputs, factor, scope=None)
models/network_ops.py:225
↓ 1 callersMethodtransfer_styles
transfer the content image by style images Args: inputs: input images [batch_size, height, width, channel] styles: a
models/avatar_net.py:67
↓ 1 callersFunctionupsample
(inputs, factor, scope=None)
models/network_ops.py:232
↓ 1 callersFunctionzca_colorization
(normalized_features, colorization_kernel, mean_features)
models/avatar_net.py:406
↓ 1 callersFunctionzca_normalization
(features)
models/avatar_net.py:373
Method__init__
(self)
datasets/dataset_utils.py:112
Method__init__
(self)
datasets/dataset_utils.py:132
Method__init__
(self, options)
models/avatar_net.py:22
Method__init__
(self, options)
models/autoencoder.py:21
Functionadaptive_instance_normalization
adaptively transform the content feature by inverse instance normalization based on the 2nd order statistics of the style feature
models/network_ops.py:248
Functionaspect_preserving_resize
(image, smallest_side)
models/preprocessing.py:244
Functionbatch_mean_image_summation
(images, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:222
Methodbuild_train_graph
(self, inputs)
models/autoencoder.py:66
Functioncompute_approximate_style_features
(features, style_loss_layers)
models/losses.py:63
Functioncompute_approximate_style_loss
compute the approximate style loss Args: style_features: a dict of the sufficient statistics of the feature maps of the style
models/losses.py:160
Functioncompute_content_and_style_features
compute the content and style features from normalized image Args: inputs: input tensor of size [batch, height, width, channel] netwo
models/losses.py:89
Functioncompute_content_loss
compute the content loss Args: content_features: a dict of the features of the input image target_features: a dict of the features of
models/losses.py:112
Functioncompute_style_loss
compute the style loss Args: style_features: a dict of the Gram matrices of the style image target_features: a dict of the Gram m
models/losses.py:136
Functioncompute_total_variation_loss_l1
compute the total variation loss L1 norm
models/losses.py:207
Functioncompute_total_variation_loss_l2
compute the total variation loss
models/losses.py:186
Functionget_model
(filename)
models/models_factory.py:17
Functionget_split
(dataset_name, split_name, dataset_dir, file_pattern=None,
datasets/dataset_utils.py:62
Methodget_training_operations
(self, optimizer, global_step, variables_to_train=tf.trainable_variables())
models/autoencoder.py:118
Functionimage_normalization
rescale the images so that their magnitude ranging from [-1, 1]
models/preprocessing.py:234
Functionimage_to_tfexample
(image_data, image_format, image_shape, image_filename)
datasets/dataset_utils.py:32
Functionlrelu
customized leaky ReLU activation function https://github.com/tensorflow/tensorflow/issues/4079
models/network_ops.py:79
Functionmain
(_)
train_image_reconstruction.py:294
Functionmain
(_)
evaluate_style_transfer.py:68
Functionmain
(_)
datasets/convert_mscoco_to_tfexamples.py:145
Functionmean_image_summation
works for one single image with dynamic shapes
models/preprocessing.py:201
Functionresidual_block_downsample
Residual block version 2 for downsampling, with preactivation Args: inputs: a tensor of size [batch, height, width, channel] dept
models/network_ops.py:128
Functionresidual_block_upsample
Residual block version 2 for upsampling, with preactivation Args: inputs: a tensor of size [batch, height, width, channel] depth:
models/network_ops.py:177
Functionresidual_unit_v0
Residual block version 0, the input and output has the same depth Args: inputs: a tensor of size [batch, height, width, channel] dept
models/network_ops.py:98
Functionvgg_16
Oxford Net VGG 16-Layers version D Example. Note: All the fully_connected layers have been transformed to conv2d layers. To use in classi
models/vgg.py:127
Functionvgg_19
Oxford Net VGG 19-Layers version E Example. Note: All the fully_connected layers have been transformed to conv2d layers. To use in classi
models/vgg.py:195
Functionvgg_a
Oxford Net VGG 11-Layers version A Example. Note: All the fully_connected layers have been transformed to conv2d layers. To use in classi
models/vgg.py:66
Functionvgg_arg_scope
Defines the VGG arg scope. Args: weight_decay: The l2 regularization coefficient. Returns: An arg_scope.
models/vgg.py:49
Functionvgg_combined_decoder
construct the decoder network with additional feature combination Args: inputs: input features [batch_size, height, width, channel]
models/vgg_decoder.py:100
Functionvgg_decoder
construct the decoder network for the vgg models Args: inputs: input features [batch_size, height, width, channel] network_name:
models/vgg_decoder.py:63
Functionvgg_decoder_arg_scope
(weight_decay=0.0005)
models/vgg_decoder.py:52
Functionvgg_multiple_combined_decoder
construct the decoder network with additional feature combination Args: inputs: input features [batch_size, height, width, channel]
models/vgg_decoder.py:147
Functionwhitening_colorization_transform
transform the content feature based on the whitening and colorization transform
models/network_ops.py:257
Functionwrite_dataset_meta_data
(dataset_dir, dataset_meta_data, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:41