Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/PanJinquan/tensorflow_models_learning
/ functions
Functions
624 in github.com/PanJinquan/tensorflow_models_learning
⨍
Functions
624
◇
Types & classes
42
↓ 1 callers
Function
subsample
Subsamples the input along the spatial dimensions. Args: inputs: A `Tensor` of size [batch, height_in, width_in, channels]. factor: The sub
slim/nets/resnet_utils.py:59
↓ 1 callers
Function
train
:param train_record_file: 训练的tfrecord文件 :param train_log_step: 显示训练过程log信息间隔 :param train_param: train参数 :param val_record_file: 验证的t
mobilenet_train_val.py:109
↓ 1 callers
Function
train
:param train_record_file: 训练的tfrecord文件 :param train_log_step: 显示训练过程log信息间隔 :param train_param: train参数 :param val_record_file: 验证的t
resnet_v1_train_val.py:109
↓ 1 callers
Function
train
:param train_record_file: 训练的tfrecord文件 :param train_log_step: 显示训练过程log信息间隔 :param train_param: train参数 :param val_record_file: 验证的t
inception_v3_train_val.py:107
↓ 1 callers
Function
train
:param train_record_file: 训练的tfrecord文件 :param train_log_step: 显示训练过程log信息间隔 :param train_param: train参数 :param val_record_file: 验证的t
vgg_train_val.py:46
↓ 1 callers
Function
train
:param train_record_file: 训练的tfrecord文件 :param train_log_step: 显示训练过程log信息间隔 :param train_param: train参数 :param val_record_file: 验证的t
inception_v1_train_val.py:107
↓ 1 callers
Function
train_model
Trains mobilenet_v1.
slim/nets/mobilenet_v1_train.py:189
↓ 1 callers
Function
write_frozen_graph
Writes frozen graph to disk. Args: frozen_graph_path: Path to write inference graph. frozen_graph_def: tf.GraphDef holding frozen
tools/exporter.py:199
↓ 1 callers
Function
write_graph_and_checkpoint
Writes the graph and the checkpoint into disk.
tools/exporter.py:259
↓ 1 callers
Function
write_saved_model
Writes SavedModel to disk. If checkpoint_path is not None bakes the weights into the graph thereby eliminating the need of checkpoint files d
tools/exporter.py:211
Function
BatchNormClassifier
(inputs, labels, scope=None, reuse=None)
slim/deployment/model_deploy_test.py:158
Function
LogisticClassifier
(inputs, labels, scope=None, reuse=None)
slim/deployment/model_deploy_test.py:149
Method
ModelFn
()
slim/deployment/model_deploy_test.py:538
Method
__call__
Runs the conv cell.
slim/nets/nasnet/nasnet_utils.py:302
Method
__enter__
(self)
slim/nets/resnet_v1.py:71
Method
__enter__
(self)
slim/nets/mobilenet/mobilenet.py:120
Method
__exit__
(self, exc_type, exc_value, traceback)
slim/nets/resnet_v1.py:74
Method
__exit__
(self, exc_type, exc_value, traceback)
slim/nets/mobilenet/mobilenet.py:123
Method
__init__
(self, num_conv_filters, drop_path_keep_prob, total_num_cells, total_training_steps)
slim/nets/nasnet/pnasnet.py:261
Method
__init__
(self, num_conv_filters, drop_path_keep_prob, total_num_cells, total_training_steps)
slim/nets/nasnet/nasnet_utils.py:456
Method
__init__
(self, num_conv_filters, drop_path_keep_prob, total_num_cells, total_training_steps)
slim/nets/nasnet/nasnet_utils.py:481
Method
__init__
(self)
slim/datasets/download_and_convert_flowers.py:55
Method
__init__
(self)
slim/datasets/build_imagenet_data.py:232
Method
__init__
Create a DeploymentConfig. The config describes how to deploy a model across multiple clones and replicas. The model will be replicated `num
slim/deployment/model_deploy.py:484
Method
__init__
(self, device, tasks)
slim/deployment/model_deploy.py:656
Function
_download_and_uncompress_dataset
Downloads cifar10 and uncompresses it locally. Args: dataset_dir: The directory where the temporary files are stored.
slim/datasets/download_and_convert_cifar10.py:124
Function
_encoded_image_string_tensor_input_placeholder
Returns input that accepts a batch of PNG or JPEG strings. Returns: A tuple of input placeholder and the output decoded images.
tools/exporter.py:139
Function
_image_tensor_input_placeholder
Returns input placeholder and a 4-D uint8 image tensor.
tools/exporter.py:130
Function
_process_image_files_batch
Processes and saves list of images as TFRecord in 1 thread. Args: coder: instance of ImageCoder to provide TensorFlow image coding utils. t
slim/datasets/build_imagenet_data.py:341
Function
_progress
(count, block_size, total_size)
slim/datasets/download_and_convert_mnist.py:159
Function
_progress
(count, block_size, total_size)
slim/datasets/dataset_utils.py:90
Function
_progress
(count, block_size, total_size)
slim/datasets/download_and_convert_cifar10.py:134
Function
alexnet_v2
AlexNet version 2. Described in: http://arxiv.org/pdf/1404.5997v2.pdf Parameters from: github.com/akrizhevsky/cuda-convnet2/blob/master/layers/
slim/nets/alexnet.py:55
Function
alexnet_v2_arg_scope
(weight_decay=0.0005)
slim/nets/alexnet.py:45
Function
apply_activation
(x, name=None, activation_fn=None)
slim/nets/mobilenet/mobilenet.py:32
Function
block17
Builds the 17x17 resnet block.
slim/nets/inception_resnet_v2.py:59
Function
block35
Builds the 35x35 resnet block.
slim/nets/inception_resnet_v2.py:33
Function
bottleneck
Bottleneck residual unit variant with BN after convolutions. This is the original residual unit proposed in [1]. See Fig. 1(a) of [2] for its def
slim/nets/resnet_v1.py:79
Function
bottleneck
Bottleneck residual unit variant with BN before convolutions. This is the full preactivation residual unit variant proposed in [2]. See Fig. 1(b)
slim/nets/resnet_v2.py:62
Function
build_nasnet_cifar
Build NASNet model for the Cifar Dataset.
slim/nets/nasnet/nasnet.py:287
Function
build_nasnet_large
Build NASNet Large model for the ImageNet Dataset.
slim/nets/nasnet/nasnet.py:384
Function
build_nasnet_mobile
Build NASNet Mobile model for the ImageNet Dataset.
slim/nets/nasnet/nasnet.py:333
Function
build_pnasnet_large
Build PNASNet Large model for the ImageNet Dataset.
slim/nets/nasnet/pnasnet.py:161
Function
build_pnasnet_mobile
Build PNASNet Mobile model for the ImageNet Dataset.
slim/nets/nasnet/pnasnet.py:208
Function
calc_reduction_layers
Figure out what layers should have reductions.
slim/nets/nasnet/nasnet_utils.py:45
Method
choose
(self, op)
slim/deployment/model_deploy.py:661
Function
cifarnet
Creates a variant of the CifarNet model. Note that since the output is a set of 'logits', the values fall in the interval of (-infinity, infinity
slim/nets/cifarnet.py:28
Function
cifarnet_arg_scope
Defines the default cifarnet argument scope. Args: weight_decay: The weight decay to use for regularizing the model. Returns: An `arg_sc
slim/nets/cifarnet.py:98
Function
clone_fn
Allows data parallelism by creating multiple clones of network_fn.
slim/train_image_classifier.py:462
Method
clone_on_cpu
(self)
slim/deployment/model_deploy.py:541
Function
conv2d_same
Strided 2-D convolution with 'SAME' padding. When stride > 1, then we do explicit zero-padding, followed by conv2d with 'VALID' padding. Note
slim/nets/resnet_utils.py:77
Function
cyclegan_arg_scope
Returns a default argument scope for all generators and discriminators. Args: instance_norm_center: Whether instance normalization applies cent
slim/nets/cyclegan.py:27
Function
cyclegan_generator_resnet
Defines the cyclegan resnet network architecture. As closely as possible following https://github.com/junyanz/CycleGAN/blob/master/models/archite
slim/nets/cyclegan.py:125
Function
cyclegan_upsample
Upsamples the given inputs. Args: net: A Tensor of size [batch_size, height, width, filters]. num_outputs: The number of output filters.
slim/nets/cyclegan.py:66
Function
decode
(encoded_image_string_tensor)
tools/exporter.py:150
Function
deploy
Deploys a Slim-constructed model across multiple clones. The deployment options are specified by the config object and support deploying one or s
slim/deployment/model_deploy.py:312
Function
depth_multiplier
(output_params, multiplier, divisible_by=8, min
slim/nets/mobilenet/mobilenet.py:97
Function
discriminator
Discriminator network for DCGAN. Construct discriminator network from inputs to the final endpoint. Args: inputs: A tensor of size [batch_si
slim/nets/dcgan.py:41
Function
disp_records
解析record文件,并显示show_nums张图片,主要用于验证生成record文件是否成功 :param tfrecord_file: record文件路径 :return:
create_tf_record.py:235
Function
download_and_uncompress_tarball
Downloads the `tarball_url` and uncompresses it locally. Args: tarball_url: The URL of a tarball file. dataset_dir: The directory where the
slim/datasets/dataset_utils.py:80
Function
expanded_conv
Depthwise Convolution Block with expansion. Builds a composite convolution that has the following structure expansion (1x1) -> depthwise (kernel_
slim/nets/mobilenet/conv_blocks.py:163
Function
export_inference_graph
Exports inference graph for the desired graph. Args: input_type: Type of input for the graph. Can be one of ['image_tensor',
tools/exporter.py:311
Function
float_feature
Returns a TF-Feature of floats. Args: values: A scalar of list of values. Returns: A TF-Feature.
slim/datasets/dataset_utils.py:56
Function
float_list_feature
(value)
create_tf_record.py:29
Function
freeze_graph2
:param input_checkpoint: :param output_graph: PB模型保存路径 :return:
convert_pb.py:88
Function
generator
Generator network for DCGAN. Construct generator network from inputs to the final endpoint. Args: inputs: A tensor with any size N. [batch_s
slim/nets/dcgan.py:109
Function
get_dataset
Given a dataset name and a split_name returns a Dataset. Args: name: String, the name of the dataset. split_name: A train/test split name.
slim/datasets/dataset_factory.py:34
Function
get_network_fn
Returns a network_fn such as `logits, end_points = network_fn(images)`. Args: name: The name of the network. num_classes: The number of cla
slim/nets/nets_factory.py:109
Function
get_preprocessing
Returns preprocessing_fn(image, height, width, **kwargs). Args: name: The name of the preprocessing function. is_training: `True` if the mo
slim/preprocessing/preprocessing_factory.py:31
Function
get_split
Gets a dataset tuple with instructions for reading ImageNet. Args: split_name: A train/test split name. dataset_dir: The base directory of
slim/datasets/imagenet.py:122
Function
get_split
Gets a dataset tuple with instructions for reading cifar10. Args: split_name: A train/test split name. dataset_dir: The base directory of t
slim/datasets/cifar10.py:44
Function
get_split
Gets a dataset tuple with instructions for reading flowers. Args: split_name: A train/validation split name. dataset_dir: The base director
slim/datasets/flowers.py:44
Function
get_split
Gets a dataset tuple with instructions for reading MNIST. Args: split_name: A train/test split name. dataset_dir: The base directory of the
slim/datasets/mnist.py:44
Function
global_avg_pool
Average pool away the height and width spatial dimensions of x.
slim/nets/nasnet/nasnet_utils.py:75
Function
has_labels
Specifies whether or not the dataset directory contains a label map file. Args: dataset_dir: The directory in which the labels file is found.
slim/datasets/dataset_utils.py:117
Function
image_to_tfexample
(image_data, image_format, height, width, class_id)
slim/datasets/dataset_utils.py:70
Function
inception_arg_scope
Defines the default arg scope for inception models. Args: weight_decay: The weight decay to use for regularizing the model. use_batch_norm:
slim/nets/inception_utils.py:32
Function
inception_resnet_v2
Creates the Inception Resnet V2 model. Args: inputs: a 4-D tensor of size [batch_size, height, width, 3]. Dimension batch_size may be und
slim/nets/inception_resnet_v2.py:291
Function
inception_resnet_v2_arg_scope
Returns the scope with the default parameters for inception_resnet_v2. Args: weight_decay: the weight decay for weights variables. batch_no
slim/nets/inception_resnet_v2.py:368
Function
inception_v1
Defines the Inception V1 architecture. This architecture is defined in: Going deeper with convolutions Christian Szegedy, Wei Liu, Yangqin
slim/nets/inception_v1.py:257
Function
inception_v2
Inception v2 model for classification. Constructs an Inception v2 network for classification as described in http://arxiv.org/abs/1502.03167.
slim/nets/inception_v2.py:452
Function
inception_v3
Inception model from http://arxiv.org/abs/1512.00567. "Rethinking the Inception Architecture for Computer Vision" Christian Szegedy, Vincent Van
slim/nets/inception_v3.py:419
Function
inception_v4
Creates the Inception V4 model. Args: inputs: a 4-D tensor of size [batch_size, height, width, 3]. num_classes: number of predicted classes
slim/nets/inception_v4.py:257
Function
init_fn
(sess)
slim/nets/mobilenet_v1_train.py:178
Method
input_and_output_same_shape_kernel3
(self)
slim/nets/cyclegan_test.py:66
Method
input_and_output_same_shape_kernel4
(self)
slim/nets/cyclegan_test.py:69
Method
input_and_output_same_shape_kernel5
(self)
slim/nets/cyclegan_test.py:72
Method
input_and_output_same_shape_kernel6
(self)
slim/nets/cyclegan_test.py:75
Function
lenet
Creates a variant of the LeNet model. Note that since the output is a set of 'logits', the values fall in the interval of (-infinity, infinity).
slim/nets/lenet.py:26
Function
lenet_arg_scope
Defines the default lenet argument scope. Args: weight_decay: The weight decay to use for regularizing the model. Returns: An `arg_scope
slim/nets/lenet.py:83
Function
main
(_)
slim/export_inference_graph.py:102
Function
main
(_)
slim/eval_image_classifier.py:85
Function
main
(_)
slim/train_image_classifier.py:390
Function
main
(_)
slim/download_and_convert_data.py:56
Function
main
(unused_arg)
slim/nets/mobilenet_v1_eval.py:147
Function
main
(unused_arg)
slim/nets/mobilenet_v1_train.py:207
Function
main
(unused_argv)
slim/datasets/build_imagenet_data.py:684
Function
mobilenet
Mobilenet model for classification, supports both V1 and V2. Note: default mode is inference, use mobilenet.training_scope to create training net
slim/nets/mobilenet/mobilenet.py:305
Function
mobilenet_base
Creates base of the mobilenet (no pooling and no logits) .
slim/nets/mobilenet/mobilenet_v2.py:175
Function
mobilenet_v1
Mobilenet v1 model for classification. Args: inputs: a tensor of shape [batch_size, height, width, channels]. num_classes: number of predic
slim/nets/mobilenet_v1.py:306
← previous
next →
201–300 of 624, ranked by callers