Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Huage001/AdaAttN
/ functions
Functions
138 in github.com/Huage001/AdaAttN
⨍
Functions
138
◇
Types & classes
23
↳
Endpoints
1
Method
forward
(self, content, style, content_key, style_key, seed=None, content_mask=None, style_mask=None)
user_specify_demo.py:66
Method
forward
(self, content4_1, style4_1, content5_1, style5_1, content4_1_key, style4_1_key, content5_1_ke
user_specify_demo.py:119
Method
forward
(self, cs, c_adain_3_feat)
user_specify_demo.py:166
Method
forward
(self, content, style, content_key, style_key, seed=None)
inference_frame.py:51
Method
forward
(self, content, style, content_key, style_key, seed=None)
inference_frame.py:91
Method
forward
(self, cs)
inference_frame.py:155
Method
forward
(self, content, style, content_key, style_key, seed=None)
models/networks.py:123
Method
forward
(self, content4_1, style4_1, content5_1, style5_1, content4_1_key, style4_1_key, content5_1_ke
models/networks.py:163
Method
forward
(self, cs, c_adain_3_feat=None)
models/networks.py:208
Function
get_option_setter
Return the static method <modify_commandline_options> of the dataset class.
data/__init__.py:41
Function
get_option_setter
Return the static method <modify_commandline_options> of the model class.
models/__init__.py:48
Function
get_params
(opt, size)
data/base_dataset.py:58
Function
get_scheduler
Return a learning rate scheduler Parameters: optimizer -- the optimizer of the network opt (option class) -- stores all
models/networks.py:8
Function
init_func
(m)
models/networks.py:51
Function
init_net
Initialize a network: 1. register CPU/GPU device (with multi-GPU support); 2. initialize the network weights Parameters: net (network)
models/networks.py:75
Method
initialize
(self, parser)
options/train_options.py:10
Method
initialize
(self, parser)
options/test_options.py:10
Function
lambda_rule
(epoch)
models/networks.py:22
Method
load_network
(self, net, path)
models/base_model.py:185
Function
mkdirs
create empty directories if they don't exist Parameters: paths (str list) -- a list of directory paths
util/util.py:78
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
data/base_dataset.py:28
Method
modify_commandline_options
Add new model-specific options, and rewrite default values for existing options. Parameters: parser -- original option p
models/base_model.py:47
Method
modify_commandline_options
(parser, is_train=True)
models/adaattn_model.py:11
Function
mouse_content_click
(event, x, y, flags, param)
user_specify_demo.py:292
Function
mouse_content_paint
(event, x, y, flags, param)
user_specify_demo.py:304
Function
mouse_style_click
(event, x, y, flags, param)
user_specify_demo.py:385
Function
mouse_style_paint
(event, x, y, flags, param)
user_specify_demo.py:397
Method
optimize_parameters
(self)
models/adaattn_model.py:219
Method
predict
(self, content, style)
predict.py:46
Function
print_numpy
Print the mean, min, max, median, std, and size of a numpy array Parameters: val (bool) -- if print the values of the numpy array
util/util.py:62
Method
query
Return an image from the pool. Parameters: images: the latest generated images from the generator Returns images from th
util/image_pool.py:23
Method
reset
Reset the self.saved status
util/visualizer.py:87
Function
save_image
Save a numpy image to the disk Parameters: image_numpy (numpy array) -- input numpy array image_path (str) -- the path o
util/util.py:49
Method
set_input
(self, input_dict)
models/adaattn_model.py:128
Method
set_requires_grad
Set requies_grad=Fasle for all the networks to avoid unnecessary computations Parameters: nets (network list) -- a list of netwo
models/base_model.py:239
Method
setup
Load and print networks; create schedulers Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subcl
models/base_model.py:78
Function
tensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
util/util.py:9
Method
validation
(self)
models/base_model.py:107
← previous
101–138 of 138, ranked by callers