Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LucasSheng/avatar-net
/ functions
Functions
91 in github.com/LucasSheng/avatar-net
⨍
Functions
91
◇
Types & classes
4
↓ 6 callers
Function
conv2d_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 callers
Function
_aspect_preserving_resize
(image, smallest_side)
models/preprocessing.py:150
↓ 3 callers
Function
_convert_dataset
(split_name, image_filenames, dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:65
↓ 3 callers
Function
bytes_feature
(values)
datasets/dataset_utils.py:28
↓ 2 callers
Function
_crop
(image, offset_height, offset_width, crop_height, crop_width)
models/preprocessing.py:19
↓ 2 callers
Function
_get_dataset_filename
(dataset_dir, split_name, shard_id)
datasets/convert_mscoco_to_tfexamples.py:59
↓ 2 callers
Function
_mean_image_subtraction
(image, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:122
↓ 2 callers
Function
conv2d_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 callers
Method
decode_image
(self, sess, image_data)
datasets/dataset_utils.py:137
↓ 2 callers
Function
get_image_filenames
helper fn that provides the full image filenames from the dataset_dir
evaluate_style_transfer.py:44
↓ 2 callers
Function
image_reader
help fn that provides numpy image coding utilities
evaluate_style_transfer.py:53
↓ 2 callers
Function
project_features
(features, projection_module='ZCA')
models/avatar_net.py:278
↓ 1 callers
Function
_central_crop
(image_list, crop_height, crop_width)
models/preprocessing.py:108
↓ 1 callers
Function
_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 callers
Function
_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 callers
Function
_dataset_exists
(dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:95
↓ 1 callers
Function
_get_filenames
(dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:28
↓ 1 callers
Function
_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 callers
Function
_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 callers
Function
_random_crop
(image_list, crop_height, crop_width)
models/preprocessing.py:45
↓ 1 callers
Function
_smallest_size_at_least
(height, width, smallest_side)
models/preprocessing.py:135
↓ 1 callers
Function
adain_colorization
(normalized_features, colorization_kernels, mean_features)
models/avatar_net.py:424
↓ 1 callers
Function
adain_normalization
(features)
models/avatar_net.py:416
↓ 1 callers
Method
auto_encoder
(self, inputs, content_layer=2, reuse=True)
models/autoencoder.py:46
↓ 1 callers
Function
batch_mean_image_subtraction
(images, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:210
↓ 1 callers
Method
build_train_graph
build the training graph for the training of the hierarchical autoencoder
models/avatar_net.py:158
↓ 1 callers
Function
compute_content_features
compute the content features from the end_point dict
models/losses.py:42
↓ 1 callers
Function
compute_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 callers
Function
compute_style_features
compute the style features from the end_point dict
models/losses.py:52
↓ 1 callers
Function
compute_sufficient_statistics
compute the gram matrix for a layer of feature
models/losses.py:34
↓ 1 callers
Method
decode_image
(self, sess, image_data)
datasets/dataset_utils.py:120
↓ 1 callers
Function
extract_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 callers
Method
get_training_operations
(self, optimizer, global_step,
models/avatar_net.py:211
↓ 1 callers
Function
has_dataset_meta_data_file
(dataset_dir, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:49
↓ 1 callers
Method
hierarchical_autoencoder
hierarchical autoencoder for content reconstruction
models/avatar_net.py:135
↓ 1 callers
Function
imsave
(filename, img)
evaluate_style_transfer.py:63
↓ 1 callers
Function
instance_norm
(inputs, epsilon=1e-10)
models/network_ops.py:90
↓ 1 callers
Function
int64_feature
(values)
datasets/dataset_utils.py:22
↓ 1 callers
Function
mean_image_subtraction
works for one single image with dynamic shapes
models/preprocessing.py:192
↓ 1 callers
Function
nearest_patch_swapping
(content_features, style_features, patch_size=3)
models/avatar_net.py:299
↓ 1 callers
Function
preprocessing_for_eval
(image, output_height, output_width, resize_side)
models/preprocessing.py:173
↓ 1 callers
Function
preprocessing_for_train
(image, output_height, output_width, resize_side)
models/preprocessing.py:165
↓ 1 callers
Function
preprocessing_image
(image, output_height, output_width, resize_side=_RESIZE_SIDE_MIN, is_training=False)
models/preprocessing.py:181
↓ 1 callers
Function
read_dataset_meta_data
(dataset_dir, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:53
↓ 1 callers
Method
read_image_dims
(self, sess, image_data)
datasets/dataset_utils.py:116
↓ 1 callers
Function
reconstruct_features
(projected_features, feature_kernels, mean_features,
models/avatar_net.py:287
↓ 1 callers
Function
run
(input_dataset_dir, output_dataset_dir)
datasets/convert_mscoco_to_tfexamples.py:105
↓ 1 callers
Function
style_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 callers
Function
subsample
(inputs, factor, scope=None)
models/network_ops.py:225
↓ 1 callers
Method
transfer_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 callers
Function
upsample
(inputs, factor, scope=None)
models/network_ops.py:232
↓ 1 callers
Function
zca_colorization
(normalized_features, colorization_kernel, mean_features)
models/avatar_net.py:406
↓ 1 callers
Function
zca_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
Function
adaptive_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
Function
aspect_preserving_resize
(image, smallest_side)
models/preprocessing.py:244
Function
batch_mean_image_summation
(images, means=(_R_MEAN, _G_MEAN, _B_MEAN))
models/preprocessing.py:222
Method
build_train_graph
(self, inputs)
models/autoencoder.py:66
Function
compute_approximate_style_features
(features, style_loss_layers)
models/losses.py:63
Function
compute_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
Function
compute_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
Function
compute_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
Function
compute_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
Function
compute_total_variation_loss_l1
compute the total variation loss L1 norm
models/losses.py:207
Function
compute_total_variation_loss_l2
compute the total variation loss
models/losses.py:186
Function
get_model
(filename)
models/models_factory.py:17
Function
get_split
(dataset_name, split_name, dataset_dir, file_pattern=None,
datasets/dataset_utils.py:62
Method
get_training_operations
(self, optimizer, global_step, variables_to_train=tf.trainable_variables())
models/autoencoder.py:118
Function
image_normalization
rescale the images so that their magnitude ranging from [-1, 1]
models/preprocessing.py:234
Function
image_to_tfexample
(image_data, image_format, image_shape, image_filename)
datasets/dataset_utils.py:32
Function
lrelu
customized leaky ReLU activation function https://github.com/tensorflow/tensorflow/issues/4079
models/network_ops.py:79
Function
main
(_)
train_image_reconstruction.py:294
Function
main
(_)
evaluate_style_transfer.py:68
Function
main
(_)
datasets/convert_mscoco_to_tfexamples.py:145
Function
mean_image_summation
works for one single image with dynamic shapes
models/preprocessing.py:201
Function
residual_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
Function
residual_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
Function
residual_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
Function
vgg_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
Function
vgg_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
Function
vgg_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
Function
vgg_arg_scope
Defines the VGG arg scope. Args: weight_decay: The l2 regularization coefficient. Returns: An arg_scope.
models/vgg.py:49
Function
vgg_combined_decoder
construct the decoder network with additional feature combination Args: inputs: input features [batch_size, height, width, channel]
models/vgg_decoder.py:100
Function
vgg_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
Function
vgg_decoder_arg_scope
(weight_decay=0.0005)
models/vgg_decoder.py:52
Function
vgg_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
Function
whitening_colorization_transform
transform the content feature based on the whitening and colorization transform
models/network_ops.py:257
Function
write_dataset_meta_data
(dataset_dir, dataset_meta_data, filename=_META_DATA_FILENAME)
datasets/dataset_utils.py:41