MCPcopy Create free account

hub / github.com/KichangKim/DeepDanbooru / functions

Functions73 in github.com/KichangKim/DeepDanbooru

⨍ Functions 73◇ Types & classes 1↳ Endpoints 5
↓ 5 callersFunctionresnet_original_bottleneck_model
https://github.com/Microsoft/CNTK/blob/master/Examples/Image/Classification/ResNet/Python/resnet_models.py
deepdanbooru/model/resnet.py:46
↓ 2 callersFunctionconv
( x, filters, kernel_size, strides=(1, 1), padding="same", initializer="he_normal" )
deepdanbooru/model/layers/__init__.py:4
↓ 2 callersFunctionexport_model_as_float32
Hotfix for exporting mixed precision model as float32.
deepdanbooru/commands/train_project.py:12
↓ 2 callersFunctiongrad_cam_test
(model, x, some_variable)
deepdanbooru/gradcam.py:21
↓ 2 callersFunctionload_image_for_evaluate
( input_: Union[str, six.BytesIO], width: int, height: int, normalize: bool = True )
deepdanbooru/data/__init__.py:13
↓ 2 callersFunctionresnet_bottleneck_inc_block
( x, output_filters, inter_filters, strides1x1=(1, 1), strides2x2=(2, 2), se=False )
deepdanbooru/model/resnet.py:26
↓ 1 callersFunctionatoi
(text)
deepdanbooru/extra/__init__.py:4
↓ 1 callersFunctionconv_bn
( x, filters, kernel_size, strides=(1, 1), padding="same", initializer="he_norma
deepdanbooru/model/layers/__init__.py:19
↓ 1 callersFunctiondownload_category_tags
( category, minimum_post_count, limit, username, api_key, page_size=1000, o
deepdanbooru/commands/download_tags.py:9
↓ 1 callersFunctionevaluate_image
( image_input: Union[str, six.BytesIO], model: Any, tags: List[str], threshold: float )
deepdanbooru/commands/evaluate.py:21
↓ 1 callersFunctionfilter_grads
(grads)
deepdanbooru/commands/grad_cam.py:27
↓ 1 callersMethodget_dataset
(self, minibatch_size)
deepdanbooru/data/dataset_wrapper.py:26
↓ 1 callersFunctionget_file_paths_in_directory
(path, patterns)
deepdanbooru/io/__init__.py:21
↓ 1 callersFunctionget_gradient
(model, x, output_mask)
deepdanbooru/commands/grad_cam.py:11
↓ 1 callersFunctionmain
AI based multi-label girl image classification system, implemented by using TensorFlow.
deepdanbooru/__main__.py:15
↓ 1 callersFunctionnorm_clip_grads
(grads)
deepdanbooru/commands/grad_cam.py:19
↓ 1 callersFunctionresnet_bottleneck_block
( x, output_filters, inter_filters, activation=True, se=False )
deepdanbooru/model/resnet.py:6
↓ 1 callersFunctionrun_test
()
deepdanbooru/gradcam.py:32
↓ 1 callersFunctionsave_txt_file
(txt_path, list)
deepdanbooru/commands/evaluate.py:9
↓ 1 callersFunctionto_onehot
(length, index)
deepdanbooru/commands/grad_cam.py:31
Method__init__
( self, inputs, tags, width, height, scale_range, rotation_range, shift_range )
deepdanbooru/data/dataset_wrapper.py:15
Functionbinary_crossentropy
(epsilon=1e-7)
deepdanbooru/model/losses/__init__.py:17
Functioncalculate_image_scale
Calculate scale for image resizing while preserving aspect ratio.
deepdanbooru/image/__init__.py:7
Functionconv2tflite
(project_path, model_path, save_path, optimize_default, optimize_experimental_sparsity, verbose)
deepdanbooru/__main__.py:234
Functionconv_bn_relu
( x, filters, kernel_size, strides=(1, 1), padding="same", initializer="he_norma
deepdanbooru/model/layers/__init__.py:42
Functionconv_gap
(x, output_filters, kernel_size=(1, 1))
deepdanbooru/model/layers/__init__.py:64
Functionconvert_to_tflite_from_from_saved_model
( project_path: str, model_path: str, save_path: str, optimizations: List[tf.lite.Optimize] = [tf.lite
deepdanbooru/commands/convert_to_tflite.py:6
Functioncreate_project
(project_path)
deepdanbooru/__main__.py:27
Functioncreate_project
Create new project with default parameters.
deepdanbooru/commands/create_project.py:6
Functioncreate_resnet_152
Original ResNet-152 Model.
deepdanbooru/model/resnet.py:124
Functioncreate_resnet_custom_v1
DeepDanbooru web (until 2019/04/20) Short, wide
deepdanbooru/model/resnet.py:142
Functioncreate_resnet_custom_v2
Experimental (blazing-deep network) Deep, narrow
deepdanbooru/model/resnet.py:160
Functioncreate_resnet_custom_v3
(x, output_dim)
deepdanbooru/model/resnet.py:178
Functioncreate_resnet_custom_v4
(x, output_dim)
deepdanbooru/model/resnet.py:192
Functiondeserialize_from_json
(path, encoding="utf-8")
deepdanbooru/io/__init__.py:11
Functiondownload_tags
(path, limit, minimum_post_count, overwrite, username, api_key)
deepdanbooru/__main__.py:41
Functiondownload_tags
( project_path, limit, minimum_post_count, is_overwrite, username, api_key )
deepdanbooru/commands/download_tags.py:77
Functionevaluate
( target_paths, # I guess its this one project_path, model_path, tags_path, threshold,
deepdanbooru/__main__.py:187
Functionevaluate
( target_paths, #this project_path, model_path, tags_path, threshold, allow_gpu,
deepdanbooru/commands/evaluate.py:43
Functionevaluate_project
(project_path, target_path, threshold)
deepdanbooru/__main__.py:122
Functionevaluate_project
(project_path, target_path, threshold)
deepdanbooru/commands/evaluate_project.py:6
Functionfocal_loss
(alpha=0.25, gamma=2.0, epsilon=1e-7)
deepdanbooru/model/losses/__init__.py:4
Functionget_image_file_paths_recursive
(folder_path, patterns_string)
deepdanbooru/io/__init__.py:29
Functiongrad
(y, x)
deepdanbooru/gradcam.py:14
Functiongrad_cam
(project_path, target_path, output_path, threshold)
deepdanbooru/__main__.py:143
Functiongrad_cam
(project_path, target_path, output_path, threshold)
deepdanbooru/commands/grad_cam.py:37
Functionload_image_records
(sqlite_path, minimum_tag_count)
deepdanbooru/data/dataset.py:11
Functionload_model_from_project
(project_path, compile_model=True)
deepdanbooru/project/project.py:42
Functionload_project
(project_path)
deepdanbooru/project/project.py:26
Functionload_tags
(tags_path)
deepdanbooru/data/dataset.py:5
Functionload_tags_from_project
(project_path)
deepdanbooru/project/project.py:57
Functionloss
(y_true, y_pred)
deepdanbooru/model/losses/__init__.py:5
Functionmake_training_database
( source_path, output_path, start_id, end_id, use_deleted, chunk_size, overwrite,
deepdanbooru/__main__.py:74
Functionmake_training_database
Make sqlite database for training. Also add system tags.
deepdanbooru/commands/make_training_database.py:5
Methodmap_load_image
(self, image_path, tag_string)
deepdanbooru/data/dataset_wrapper.py:43
Methodmap_transform_image_and_label
(self, image, tag_string)
deepdanbooru/data/dataset_wrapper.py:67
Methodmap_transform_image_and_label_py
(self, image, tag_string)
deepdanbooru/data/dataset_wrapper.py:74
Functionnatural_keys
alist.sort(key=natural_keys) sorts in human order http://nedbatchelder.com/blog/200712/human_sorting.html (See Toothy's implementation
deepdanbooru/extra/__init__.py:8
Functionnatural_sorted
(iterable)
deepdanbooru/extra/__init__.py:17
Functionpackages
()
tests/test_main.py:27
Functionrepeat_blocks
(x, block_delegate, count, **kwargs)
deepdanbooru/model/layers/__init__.py:71
Functionresnet_longterm_bottleneck_model
Add long-term shortcut.
deepdanbooru/model/resnet.py:80
Functionserialize_as_json
(target_object, path, encoding="utf-8")
deepdanbooru/io/__init__.py:6
Functionsqueeze_excitation
Squeeze-Excitation layer from https://arxiv.org/abs/1709.01507
deepdanbooru/model/layers/__init__.py:79
Functiontest_help
(func_name)
tests/test_main.py:18
Functiontest_import
()
tests/test_main.py:13
Functiontest_load_image_for_evaluate
(tmp_path, use_bytes_io)
tests/test_main.py:67
Functiontest_package_setup
(packages)
tests/test_main.py:32
Functiontest_readme_pkg
(packages)
tests/test_main.py:45
Functiontrain_project
(project_path, source_model)
deepdanbooru/__main__.py:105
Functiontrain_project
(project_path, source_model)
deepdanbooru/commands/train_project.py:27
Functiontransform_and_pad_image
Transform image and pad by edge pixles.
deepdanbooru/image/__init__.py:25
Functiontry_create_directory
(path)
deepdanbooru/io/__init__.py:16