Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/azadis/CompositionalGAN
/ functions
Functions
215 in github.com/azadis/CompositionalGAN
⨍
Functions
215
◇
Types & classes
36
↓ 20 callers
Method
save
(self)
util/html.py:45
↓ 10 callers
Method
__init__
(self, use_lsgan=True, target_real_label=1.0, target_fake_label=0.0, tensor=torch.FloatTensor
models/networks.py:239
↓ 10 callers
Method
query
(self, images)
util/image_pool.py:14
↓ 8 callers
Function
CreateDataLoader
(opt)
data/data_loader.py:2
↓ 8 callers
Method
calc_gradient_penalty
Gradient Penalty
models/objCompose_unsupervised_model.py:484
↓ 8 callers
Method
display_current_results
(self, visuals, epoch, save_result, stepsize=1,im_name='',n_latest=50)
util/visualizer.py:37
↓ 8 callers
Method
get_current_visuals
(self)
models/AFN_model.py:174
↓ 8 callers
Method
load_data
()
data/base_data_loader.py:10
↓ 7 callers
Function
create_model
(opt)
models/models.py:8
↓ 7 callers
Method
set_input
(self, input)
models/AFN_model.py:58
↓ 6 callers
Method
calc_gradient_penalty
Gradient Penalty
models/objCompose_supervised_model.py:298
↓ 6 callers
Method
parse
(self)
options/base_options.py:71
↓ 6 callers
Method
plot_current_errors
(self, epoch, counter_ratio, opt, errors)
util/visualizer.py:100
↓ 6 callers
Method
print_current_errors
(self, epoch, i, errors, t)
util/visualizer.py:116
↓ 6 callers
Method
reset
(self)
util/visualizer.py:33
↓ 5 callers
Method
get_current_errors
(self)
models/AFN_model.py:167
↓ 5 callers
Method
load_network
(self, network, network_label, epoch_label)
models/base_model.py:68
↓ 5 callers
Method
tnsrs2ims
(self, tensor_names)
models/base_model.py:128
↓ 5 callers
Method
tocuda
(self, var_names)
models/base_model.py:121
↓ 4 callers
Method
add_header
(self, str)
util/html.py:25
↓ 4 callers
Method
add_images
(self, ims, txts, links, width=400)
util/html.py:33
↓ 4 callers
Method
get_image_paths
(self)
models/AFN_model.py:133
↓ 4 callers
Function
iou
(mask1, mask2)
util/robust_loss.py:109
↓ 4 callers
Function
overlap
(mask1, mask2)
util/robust_loss.py:102
↓ 4 callers
Method
save_images
(self, webpage, visuals, image_path, aspect_ratio=1.0)
util/visualizer.py:126
↓ 4 callers
Method
update_learning_rate
(self)
models/base_model.py:95
↓ 3 callers
Method
eval
(self)
models/base_model.py:101
↓ 3 callers
Method
forward
starting from input object images
models/objCompose_supervised_model.py:219
↓ 3 callers
Method
forward_test
starting from input object images @ test time
models/objCompose_supervised_model.py:285
↓ 3 callers
Function
init_weights
(net, init_type='normal')
models/networks.py:84
↓ 3 callers
Method
optimize_parameters
(self)
models/AFN_model.py:160
↓ 3 callers
Method
save
(self, label)
models/base_model.py:42
↓ 2 callers
Function
__scale_width
(img, target_width)
data/base_dataset.py:59
↓ 2 callers
Method
backward_D_test
backward pass for the discriminator @ test time
models/objCompose_unsupervised_model.py:691
↓ 2 callers
Method
backward_G_test
backward pass for the generator @ test time
models/objCompose_unsupervised_model.py:761
↓ 2 callers
Method
forward
starting from segments of input_B @ training time
models/objCompose_unsupervised_model.py:413
↓ 2 callers
Method
forward_AFN
(self)
models/AFN_compose_model.py:73
↓ 2 callers
Method
get_image_dir
(self)
util/html.py:22
↓ 2 callers
Function
get_norm_layer
(norm_type='instance')
models/networks.py:100
↓ 2 callers
Method
load_networks
(self)
models/base_model.py:75
↓ 2 callers
Function
mkdir
(path)
util/util.py:65
↓ 2 callers
Method
optimize_parameters_completion
(self, total_steps)
models/objCompose_unsupervised_model.py:830
↓ 2 callers
Function
printLayer
(layer, layer_name)
util/conv_info.py:39
↓ 2 callers
Method
print_networks
(self, verbose=False)
models/base_model.py:108
↓ 2 callers
Method
save_network
(self, network, network_label, epoch_label, gpu_ids)
models/base_model.py:46
↓ 2 callers
Method
save_networks
(self, model_names, epoch_labels)
models/base_model.py:53
↓ 2 callers
Method
set_input_test
samples at test time; no target composite image given
models/objCompose_supervised_model.py:160
↓ 2 callers
Method
set_optimizers
(self)
models/base_model.py:85
↓ 1 callers
Function
CreateDataset
(opt)
data/custom_dataset_data_loader.py:11
↓ 1 callers
Method
add_table
(self, border=1)
util/html.py:29
↓ 1 callers
Method
backward_AFN
(self)
models/AFN_compose_model.py:97
↓ 1 callers
Method
backward_D
backward pass for the discriminator in training the unpaired model
models/objCompose_unsupervised_model.py:612
↓ 1 callers
Method
backward_D
backward pass for the discriminator in training the paired model
models/objCompose_supervised_model.py:311
↓ 1 callers
Method
backward_D_completion
Backward pass for the discriminator in training the inpainting networks only
models/objCompose_unsupervised_model.py:534
↓ 1 callers
Method
backward_D_test
backward pass for the discriminator @ test time
models/objCompose_supervised_model.py:467
↓ 1 callers
Method
backward_G
backward pass for the generator in training the unsupervised model
models/objCompose_unsupervised_model.py:562
↓ 1 callers
Method
backward_G
backward pass for the generator in training the paired model
models/objCompose_supervised_model.py:380
↓ 1 callers
Method
backward_G_completion
Backward pass for the generator in training the inpainting networks only
models/objCompose_unsupervised_model.py:504
↓ 1 callers
Method
backward_G_test
backward pass for the generator @ test time
models/objCompose_supervised_model.py:536
↓ 1 callers
Method
backward_STN
backward pass for training STN networks only
models/objCompose_unsupervised_model.py:497
↓ 1 callers
Method
backward_flow
(self)
models/AFN_model.py:140
↓ 1 callers
Method
build_conv_block
(self, dim, padding_type, norm_layer, use_dropout, use_bias)
models/networks.py:473
↓ 1 callers
Method
forward
(self)
models/AFN_model.py:97
↓ 1 callers
Method
forward_STN
Forward pass for the spatial transformer network
models/objCompose_unsupervised_model.py:252
↓ 1 callers
Method
forward_inpainting
forward pass for the inpainting network
models/objCompose_unsupervised_model.py:268
↓ 1 callers
Method
forward_test
starting from full OBJ1 and OBJ2 @ test time
models/objCompose_unsupervised_model.py:298
↓ 1 callers
Method
get_current_errors_test
(self)
models/objCompose_supervised_model.py:673
↓ 1 callers
Method
get_current_visuals_A_segment
(self)
models/objCompose_unsupervised_model.py:974
↓ 1 callers
Method
get_current_visuals_STN
(self)
models/objCompose_unsupervised_model.py:956
↓ 1 callers
Method
get_current_visuals_completion
(self)
models/objCompose_unsupervised_model.py:965
↓ 1 callers
Method
get_target_tensor
(self, input, target_is_real)
models/networks.py:252
↓ 1 callers
Method
initialize
(self, opt)
data/custom_dataset_data_loader.py:39
↓ 1 callers
Method
initialize
(self, opt)
data/AFN_compose_dataset.py:20
↓ 1 callers
Method
initialize
(self)
options/base_options.py:19
↓ 1 callers
Method
initialize
(self)
options/baseAFN_options.py:12
↓ 1 callers
Method
initialize
(self)
options/baseAFNCompose_options.py:12
↓ 1 callers
Method
initialize
(self, opt)
models/AFN_compose_model.py:16
↓ 1 callers
Function
is_image_file
(filename)
data/image_folder.py:20
↓ 1 callers
Function
make_dataset
(dir)
data/image_folder.py:24
↓ 1 callers
Method
name
(self)
data/custom_dataset_data_loader.py:36
↓ 1 callers
Method
name
(self)
data/AFN_compose_dataset.py:99
↓ 1 callers
Method
name
(self)
models/AFN_compose_model.py:13
↓ 1 callers
Method
optimize_parameters_AFN
(self)
models/AFN_compose_model.py:111
↓ 1 callers
Method
optimize_parameters_STN
(self)
models/objCompose_unsupervised_model.py:852
↓ 1 callers
Method
optimize_parameters_test
(self, total_steps)
models/objCompose_supervised_model.py:642
↓ 1 callers
Function
outFromIn
(conv, layerIn)
util/conv_info.py:20
↓ 1 callers
Method
set_input_train
samples of real distribution (from training set) to be used at test time
models/objCompose_supervised_model.py:141
↓ 1 callers
Method
test
(self)
models/AFN_model.py:115
Function
BorderOne
(img)
data/base_dataset.py:67
Function
ChangeDirection
(path)
data/base_dataset.py:77
Function
Gaussian2D
(dim, mu_x, mu_y, sigma_x, sigma_y,gpu_ids=[])
util/helper.py:7
Method
__call__
(self, input, target_is_real)
models/networks.py:270
Method
__getitem__
(self, index)
data/compose_dataset.py:42
Method
__getitem__
(self, index)
data/compose_dataset.py:189
Method
__getitem__
(self, index)
data/AFN_compose_dataset.py:40
Method
__getitem__
(self, index)
data/image_folder.py:57
Method
__getitem__
(self, index)
data/AFN_dataset.py:52
Method
__init__
(self)
data/base_data_loader.py:3
Method
__init__
(self)
data/base_dataset.py:11
Method
__init__
(self, root, transform=None, return_paths=False, loader=default_loader)
data/image_folder.py:43
next →
1–100 of 215, ranked by callers