Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/albertpumarola/GANimation
/ functions
Functions
135 in github.com/albertpumarola/GANimation
⨍
Functions
135
◇
Types & classes
18
↓ 6 callers
Method
forward
(self, keep_data_for_visuals=False, return_estimates=False)
models/ganimation.py:123
↓ 5 callers
Method
_do_if_necessary_saturate_mask
(self, m, saturate=False)
models/ganimation.py:404
↓ 5 callers
Method
save
(self, label)
models/models.py:75
↓ 4 callers
Function
mkdir
(path)
utils/util.py:42
↓ 3 callers
Method
_compute_loss_D
(self, estim, is_real)
models/ganimation.py:310
↓ 3 callers
Method
get_by_name
(dataset_name, opt, is_for_train)
data/dataset.py:13
↓ 3 callers
Method
get_current_errors
(self)
models/models.py:69
↓ 3 callers
Method
load
(self)
models/models.py:78
↓ 3 callers
Method
plot_scalars
(self, scalars, it, is_train)
utils/tb_visualizer.py:36
↓ 3 callers
Method
set_input
(self, input)
models/models.py:44
↓ 2 callers
Method
_compute_loss_smooth
(self, mat)
models/ganimation.py:313
↓ 2 callers
Method
_get_cond_by_id
(self, id)
data/dataset_aus.py:101
↓ 2 callers
Method
_load_network
(self, network, network_label, epoch_label)
models/models.py:101
↓ 2 callers
Method
_load_optimizer
(self, optimizer, optimizer_label, epoch_label)
models/models.py:86
↓ 2 callers
Method
_save_network
(self, network, network_label, epoch_label)
models/models.py:95
↓ 2 callers
Method
_save_optimizer
(self, optimizer, optimizer_label, epoch_label)
models/models.py:81
↓ 2 callers
Method
display_current_results
(self, visuals, it, is_train, save_visuals=False)
utils/tb_visualizer.py:24
↓ 2 callers
Method
forward
(self, keep_data_for_visuals=False)
models/models.py:53
↓ 2 callers
Method
get_by_name
(model_name, *args, **kwargs)
models/models.py:10
↓ 2 callers
Method
get_current_visuals
(self)
models/models.py:66
↓ 2 callers
Method
init_weights
(self)
networks/networks.py:34
↓ 2 callers
Method
load_data
(self)
data/custom_dataset_data_loader.py:21
↓ 2 callers
Method
parse
(self)
options/base_options.py:36
↓ 2 callers
Method
set_eval
(self)
models/models.py:50
↓ 2 callers
Method
set_train
(self)
models/models.py:47
↓ 1 callers
Method
__init__
(self, conv_dim=64, c_dim=5, repeat_num=6)
networks/generator_wasserstein_gan.py:8
↓ 1 callers
Method
_create_dataset
(self)
data/custom_dataset_data_loader.py:12
↓ 1 callers
Method
_create_discriminator
(self)
models/ganimation.py:52
↓ 1 callers
Method
_create_generator
(self)
models/ganimation.py:49
↓ 1 callers
Method
_create_transform
(self)
data/dataset.py:46
↓ 1 callers
Method
_display_terminal
(self, iter_start_time, i_epoch, i_train_batch, visuals_flag)
train.py:92
↓ 1 callers
Method
_display_visualizer_train
(self, total_steps)
train.py:97
↓ 1 callers
Method
_display_visualizer_val
(self, i_epoch, total_steps)
train.py:102
↓ 1 callers
Method
_forward_D
(self)
models/ganimation.py:271
↓ 1 callers
Method
_forward_G
(self, keep_data_for_visuals)
models/ganimation.py:224
↓ 1 callers
Method
_generate_random_cond
(self)
data/dataset_aus.py:111
↓ 1 callers
Method
_get_img_by_id
(self, id)
data/dataset_aus.py:107
↓ 1 callers
Method
_get_set_gpus
(self)
options/base_options.py:80
↓ 1 callers
Method
_gradinet_penalty_D
(self, fake_imgs_masked)
models/ganimation.py:289
↓ 1 callers
Method
_img_morph
(self, img, expresion)
test.py:32
↓ 1 callers
Method
_init_create_networks
(self)
models/ganimation.py:34
↓ 1 callers
Method
_init_losses
(self)
models/ganimation.py:72
↓ 1 callers
Method
_init_prefetch_inputs
(self)
models/ganimation.py:65
↓ 1 callers
Method
_init_train_vars
(self)
models/ganimation.py:55
↓ 1 callers
Method
_morph_face
(self, face, expresion)
test.py:46
↓ 1 callers
Method
_print
(self, args)
options/base_options.py:93
↓ 1 callers
Method
_read_conds
(self, file_path)
data/dataset_aus.py:97
↓ 1 callers
Method
_read_dataset_paths
(self)
data/dataset_aus.py:61
↓ 1 callers
Method
_read_ids
(self, file_path)
data/dataset_aus.py:93
↓ 1 callers
Method
_save
(self, args)
options/base_options.py:99
↓ 1 callers
Method
_save_img
(self, img, filename)
test.py:54
↓ 1 callers
Method
_set_and_check_load_epoch
(self)
options/base_options.py:60
↓ 1 callers
Method
_train
(self)
train.py:29
↓ 1 callers
Method
_train_epoch
(self, i_epoch)
train.py:56
↓ 1 callers
Method
get_current_scalars
(self)
models/models.py:72
↓ 1 callers
Function
get_data
(filepaths)
data/prepare_au_annotations.py:14
↓ 1 callers
Method
initialize
(self)
options/base_options.py:11
↓ 1 callers
Method
load
(self)
models/ganimation.py:372
↓ 1 callers
Function
main
()
test.py:60
↓ 1 callers
Function
main
()
data/prepare_au_annotations.py:26
↓ 1 callers
Method
morph_file
(self, img_path, expresion)
test.py:26
↓ 1 callers
Method
optimize_parameters
(self)
models/models.py:63
↓ 1 callers
Method
print_current_train_errors
(self, epoch, i, iters_per_epoch, errors, t, visuals_were_stored)
utils/tb_visualizer.py:41
↓ 1 callers
Method
print_current_validate_errors
(self, epoch, errors, t)
utils/tb_visualizer.py:52
↓ 1 callers
Function
resize_face
Resize face to a given size :param face_img: cv::mat HxWx3 :param size: new H and W (size x size). 128 by default. :return: cv::mat s
utils/face_utils.py:60
↓ 1 callers
Function
save_dict
(data, name)
data/prepare_au_annotations.py:22
↓ 1 callers
Function
tensor2im
(img, imtype=np.uint8, unnormalize=True, idx=0, nrows=None)
utils/util.py:9
↓ 1 callers
Method
update_learning_rate
(self)
models/models.py:110
Method
__del__
(self)
utils/tb_visualizer.py:21
Method
__getitem__
(self, index)
data/dataset_aus.py:19
Method
__init__
(self)
train.py:11
Method
__init__
(self, opt)
test.py:17
Method
__init__
(self, opt)
utils/tb_visualizer.py:9
Method
__init__
(self)
networks/networks.py:5
Method
__init__
(self)
networks/networks.py:26
Method
__init__
(self, dim_in, dim_out)
networks/generator_wasserstein_gan.py:58
Method
__init__
(self, image_size=128, conv_dim=64, c_dim=5, repeat_num=6)
networks/discriminator_wasserstein_gan.py:7
Method
__init__
(self)
data/dataset.py:9
Method
__init__
(self, opt, is_for_train)
data/dataset.py:25
Method
__init__
(self, opt, is_for_train)
data/dataset_aus.py:12
Method
__init__
(self, opt, is_for_train=True)
data/custom_dataset_data_loader.py:6
Method
__init__
(self)
options/base_options.py:7
Method
__init__
(self)
models/models.py:6
Method
__init__
(self, opt)
models/models.py:25
Method
__init__
(self, opt)
models/ganimation.py:13
Method
__len__
(self)
data/dataset_aus.py:58
Method
__len__
(self)
data/custom_dataset_data_loader.py:24
Method
_create_transform
(self)
data/dataset_aus.py:79
Method
_get_all_files_in_subfolders
(self, dir, is_file)
data/dataset.py:58
Method
_get_norm_layer
(self, norm_type='batch')
networks/networks.py:47
Method
_get_scheduler
(self, optimizer, opt)
models/models.py:120
Method
_is_csv_file
(self, filename)
data/dataset.py:55
Method
_is_image_file
(self, filename)
data/dataset.py:52
Method
_l1_loss_with_target_gradients
(self, input, target)
models/ganimation.py:401
Method
_weights_init_fn
(self, m)
networks/networks.py:37
Function
crop_face_with_bb
Crop face in image given bb :param img: cv::mat HxWx3 :param bb: 4 (<x,y,w,h>) :return: HxWx3
utils/face_utils.py:44
Function
detect_biggest_face
Detect biggest face in image :param img: cv::mat HxWx3 RGB :return: 4 <x,y,w,h>
utils/face_utils.py:21
Function
detect_faces
Detect faces in image :param img: cv::mat HxWx3 RGB :return: yield 4 <x,y,w,h>
utils/face_utils.py:8
Function
detect_landmarks
(face_img)
utils/face_utils.py:69
Method
forward
(self, x, c)
networks/generator_wasserstein_gan.py:48
next →
1–100 of 135, ranked by callers