MCPcopy Create free account

hub / github.com/awesome-davian/Text2Colors / functions

Functions90 in github.com/awesome-davian/Text2Colors

↓ 18 callersMethod__init__
(self, imsize, add_L)
model/PCN.py:287
↓ 5 callersFunctionlab2rgb_1d
(in_lab, clip=True)
util.py:160
↓ 4 callersMethod__init__
(self, hidden_size, max_length)
model/TPN.py:94
↓ 4 callersMethodload_model
(self, mode, resume_epoch)
solver.py:132
↓ 3 callersMethodinit_hidden
(self,batch_size)
model/TPN.py:50
↓ 2 callersFunctionload_pretrained_embedding
(dictionary, embed_file, embed_dim)
util.py:34
↓ 2 callersMethodprepare_dict
(self)
solver.py:21
↓ 2 callersFunctionprocess_image
(image_data, batch_size, imsize)
util.py:77
↓ 2 callersFunctionprocess_palette_lab
(pal_data, batch_size)
util.py:102
↓ 1 callersFunctionKL_loss
(mu, logvar)
util.py:154
↓ 1 callersMethodbuild_model
(self, mode)
solver.py:53
↓ 1 callersMethodencode
(self, text_embedding)
model/TPN.py:15
↓ 1 callersMethodindex_element
(self, element)
util.py:25
↓ 1 callersMethodindex_elements
(self, data)
util.py:20
↓ 1 callersFunctionmain
(args)
main.py:8
↓ 1 callersFunctionp2c_loader
(dataset, batch_size, idx=0)
data_loader.py:91
↓ 1 callersMethodprepare_data
(self, images, palettes, always_give_global_hint, add_L)
solver.py:35
↓ 1 callersFunctionprocess_global_ab
(input_ab, batch_size, always_give_global_hint)
util.py:111
↓ 1 callersFunctionprocess_global_lab
(input_lab, batch_size, always_give_global_hint)
util.py:125
↓ 1 callersFunctionprocess_global_sampling_lab
(palette, batch_size, imsize, hist_mean, hist_std)
util.py:146
↓ 1 callersFunctionprocess_palette_ab
(pal_data, batch_size)
util.py:94
↓ 1 callersMethodreparametrize
(self, mu, logvar)
model/TPN.py:21
↓ 1 callersMethodscore
(self, hidden, encoder_output)
model/TPN.py:114
↓ 1 callersFunctiont2p_loader
(batch_size, input_dict)
data_loader.py:49
↓ 1 callersMethodtest_TPN
(self)
solver.py:366
↓ 1 callersFunctiontest_loader
(dataset, batch_size, input_dict)
data_loader.py:174
↓ 1 callersMethodtest_text2colors
(self)
solver.py:440
↓ 1 callersMethodtrain_PCN
(self)
solver.py:286
↓ 1 callersMethodtrain_TPN
(self)
solver.py:157
Method__getitem__
(self, index)
data_loader.py:40
Method__getitem__
(self, idx)
data_loader.py:87
Method__getitem__
Returns one data pair.
data_loader.py:163
Method__init__
(self, args)
solver.py:15
Method__init__
(self, src_path, trg_path, input_dict)
data_loader.py:10
Method__init__
(self, image_dir, pal_dir)
data_loader.py:73
Method__init__
(self, input_dict, txt_path, pal_path, img_path, transform=None)
data_loader.py:122
Method__init__
(self)
util.py:13
Method__init__
(self, vocab_size, embed_dim, W_emb, train_emb)
util.py:60
Method__init__
(self, in_size, out_size, kernel_size=3)
model/PCN.py:7
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:17
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:33
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:47
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:64
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:79
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:98
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:115
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:132
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:149
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu)
model/PCN.py:166
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu, space_dropout=False)
model/PCN.py:183
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu, space_dropout=False)
model/PCN.py:203
Method__init__
(self, in_size, out_size, kernel_size=3, activation=F.relu, space_dropout=False)
model/PCN.py:221
Method__init__
(self, in_size, out_size, kernel_size=1, activation=F.sigmoid, space_dropout=False)
model/PCN.py:237
Method__init__
(self, in_size, out_size, kernel_size=1, activation=F.relu, space_dropout=False)
model/PCN.py:248
Method__init__
(self, image_size, add_L)
model/PCN.py:259
Method__init__
(self, add_L, imsize, conv_dim=64, repeat_num=5)
model/PCN.py:343
Method__init__
(self)
model/TPN.py:8
Method__init__
(self, input_size, hidden_size, n_layers, dropout_p, W_emb=None)
model/TPN.py:33
Method__init__
(self, input_dict, hidden_size, n_layers=1, dropout_p=0.1)
model/TPN.py:57
Method__init__
(self, color_size=15, hidden_dim=150)
model/TPN.py:123
Method__len__
(self)
data_loader.py:45
Method__len__
(self)
data_loader.py:84
Method__len__
(self)
data_loader.py:160
Methodforward
(self, doc)
util.py:72
Methodforward
(self, x)
model/PCN.py:11
Methodforward
(self, x)
model/PCN.py:24
Methodforward
(self, x)
model/PCN.py:39
Methodforward
(self, x)
model/PCN.py:55
Methodforward
(self, x)
model/PCN.py:71
Methodforward
(self, x)
model/PCN.py:88
Methodforward
(self, x)
model/PCN.py:106
Methodforward
(self, x)
model/PCN.py:123
Methodforward
(self, x)
model/PCN.py:140
Methodforward
(self, x)
model/PCN.py:157
Methodforward
(self, x)
model/PCN.py:174
Methodforward
(self, x, bridge)
model/PCN.py:193
Methodforward
(self, x, bridge)
model/PCN.py:211
Methodforward
(self, x, bridge)
model/PCN.py:229
Methodforward
(self, x)
model/PCN.py:242
Methodforward
(self, x)
model/PCN.py:253
Methodforward
(self, x, dim)
model/PCN.py:270
Methodforward
(self, x, side_input)
model/PCN.py:311
Methodforward
(self, x)
model/PCN.py:370
Methodforward
(self, text_embedding)
model/TPN.py:26
Methodforward
(self, word_inputs, hidden)
model/TPN.py:43
Methodforward
(self, last_palette, last_decoder_hidden, encoder_outputs, each_input_size, i)
model/TPN.py:75
Methodforward
(self, hidden, encoder_outputs, each_size)
model/TPN.py:103
Methodforward
(self, color, text)
model/TPN.py:139
Functioninit_weights_normal
(m)
util.py:167
Functionprocess_global_sampling_ab
(palette, batch_size, imsize, hist_mean, hist_std)
util.py:139