Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/awesome-davian/Text2Colors
/ functions
Functions
90 in github.com/awesome-davian/Text2Colors
⨍
Functions
90
◇
Types & classes
30
↓ 18 callers
Method
__init__
(self, imsize, add_L)
model/PCN.py:287
↓ 5 callers
Function
lab2rgb_1d
(in_lab, clip=True)
util.py:160
↓ 4 callers
Method
__init__
(self, hidden_size, max_length)
model/TPN.py:94
↓ 4 callers
Method
load_model
(self, mode, resume_epoch)
solver.py:132
↓ 3 callers
Method
init_hidden
(self,batch_size)
model/TPN.py:50
↓ 2 callers
Function
load_pretrained_embedding
(dictionary, embed_file, embed_dim)
util.py:34
↓ 2 callers
Method
prepare_dict
(self)
solver.py:21
↓ 2 callers
Function
process_image
(image_data, batch_size, imsize)
util.py:77
↓ 2 callers
Function
process_palette_lab
(pal_data, batch_size)
util.py:102
↓ 1 callers
Function
KL_loss
(mu, logvar)
util.py:154
↓ 1 callers
Method
build_model
(self, mode)
solver.py:53
↓ 1 callers
Method
encode
(self, text_embedding)
model/TPN.py:15
↓ 1 callers
Method
index_element
(self, element)
util.py:25
↓ 1 callers
Method
index_elements
(self, data)
util.py:20
↓ 1 callers
Function
main
(args)
main.py:8
↓ 1 callers
Function
p2c_loader
(dataset, batch_size, idx=0)
data_loader.py:91
↓ 1 callers
Method
prepare_data
(self, images, palettes, always_give_global_hint, add_L)
solver.py:35
↓ 1 callers
Function
process_global_ab
(input_ab, batch_size, always_give_global_hint)
util.py:111
↓ 1 callers
Function
process_global_lab
(input_lab, batch_size, always_give_global_hint)
util.py:125
↓ 1 callers
Function
process_global_sampling_lab
(palette, batch_size, imsize, hist_mean, hist_std)
util.py:146
↓ 1 callers
Function
process_palette_ab
(pal_data, batch_size)
util.py:94
↓ 1 callers
Method
reparametrize
(self, mu, logvar)
model/TPN.py:21
↓ 1 callers
Method
score
(self, hidden, encoder_output)
model/TPN.py:114
↓ 1 callers
Function
t2p_loader
(batch_size, input_dict)
data_loader.py:49
↓ 1 callers
Method
test_TPN
(self)
solver.py:366
↓ 1 callers
Function
test_loader
(dataset, batch_size, input_dict)
data_loader.py:174
↓ 1 callers
Method
test_text2colors
(self)
solver.py:440
↓ 1 callers
Method
train_PCN
(self)
solver.py:286
↓ 1 callers
Method
train_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
Method
forward
(self, doc)
util.py:72
Method
forward
(self, x)
model/PCN.py:11
Method
forward
(self, x)
model/PCN.py:24
Method
forward
(self, x)
model/PCN.py:39
Method
forward
(self, x)
model/PCN.py:55
Method
forward
(self, x)
model/PCN.py:71
Method
forward
(self, x)
model/PCN.py:88
Method
forward
(self, x)
model/PCN.py:106
Method
forward
(self, x)
model/PCN.py:123
Method
forward
(self, x)
model/PCN.py:140
Method
forward
(self, x)
model/PCN.py:157
Method
forward
(self, x)
model/PCN.py:174
Method
forward
(self, x, bridge)
model/PCN.py:193
Method
forward
(self, x, bridge)
model/PCN.py:211
Method
forward
(self, x, bridge)
model/PCN.py:229
Method
forward
(self, x)
model/PCN.py:242
Method
forward
(self, x)
model/PCN.py:253
Method
forward
(self, x, dim)
model/PCN.py:270
Method
forward
(self, x, side_input)
model/PCN.py:311
Method
forward
(self, x)
model/PCN.py:370
Method
forward
(self, text_embedding)
model/TPN.py:26
Method
forward
(self, word_inputs, hidden)
model/TPN.py:43
Method
forward
(self, last_palette, last_decoder_hidden, encoder_outputs, each_input_size, i)
model/TPN.py:75
Method
forward
(self, hidden, encoder_outputs, each_size)
model/TPN.py:103
Method
forward
(self, color, text)
model/TPN.py:139
Function
init_weights_normal
(m)
util.py:167
Function
process_global_sampling_ab
(palette, batch_size, imsize, hist_mean, hist_std)
util.py:139