Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Arctanxy/W-Net-PyTorch
/ functions
Functions
47 in github.com/Arctanxy/W-Net-PyTorch
⨍
Functions
47
◇
Types & classes
19
↓ 7 callers
Method
__init__
(self, M=5, num_fonts=80, num_characters=3375)
src/model.py:84
↓ 7 callers
Function
generate_img
(word, font_file, font_size, w=80)
src/data.py:20
↓ 3 callers
Method
__init__
(self)
src/loss_func.py:33
↓ 1 callers
Function
calc_gradient_penalty
(D, x_real, x_fake, x1, x2)
src/loss_func.py:56
↓ 1 callers
Method
eval_model
(self)
src/trainer.py:266
↓ 1 callers
Method
init_model
(self)
src/trainer.py:177
↓ 1 callers
Method
save_model
(self, check_point)
src/trainer.py:240
↓ 1 callers
Method
train_one_epoch
(self, epoch)
src/trainer.py:22
Method
__getattr__
(self, key)
src/config.py:21
Method
__getitem__
(self, index)
src/preload_data.py:121
Method
__getitem__
(self, index)
src/data.py:116
Method
__init__
(self, yml_path)
src/config.py:8
Method
__init__
(self, font_size=60, transform=None, subset="train")
src/preload_data.py:28
Method
__init__
(self, data, shuffle=True)
src/preload_data.py:151
Method
__init__
(self, sampler, batch_size, drop_last)
src/preload_data.py:172
Method
__init__
Constructor for the LabelSmoothing module. :param smoothing: label smoothing factor
src/loss_func.py:13
Method
__init__
(self)
src/loss_func.py:92
Method
__init__
(self)
src/loss_func.py:169
Method
__init__
(self, font_size=60, transform=None, subset="train")
src/data.py:80
Method
__init__
(self, train_loader, valid_loader)
src/trainer.py:16
Method
__init__
(self, in_c, out_c, ks=5, pad=2, s=2, bn=False)
src/model.py:11
Method
__init__
(self, in_c, out_c, ks=6, pad=2, s=2, bn=False)
src/model.py:31
Method
__init__
(self)
src/model.py:52
Method
__init__
(self, in_c, out_c)
src/model.py:71
Method
__init__
(self, num_fonts=80, num_characters=3500 + 1)
src/model.py:141
Method
__init__
(self, num_characters=3375)
src/model.py:171
Method
__init__
(self, num_fonts=80)
src/model.py:180
Method
__iter__
(self)
main.py:25
Method
__iter__
(self)
src/preload_data.py:155
Method
__iter__
(self)
src/preload_data.py:177
Method
__len__
(self)
src/preload_data.py:146
Method
__len__
(self)
src/preload_data.py:167
Method
__len__
(self)
src/preload_data.py:201
Method
__len__
(self)
src/data.py:143
Method
__str__
(self)
src/config.py:24
Method
forward
(self, x, target)
src/loss_func.py:22
Method
forward
(self, x_fake, x_real)
src/loss_func.py:36
Method
forward
( self, out, out_real, real_label, real_style_label, cha
src/loss_func.py:99
Method
forward
( self, out_real, out_fake, real_label, fake_label, real
src/loss_func.py:176
Method
forward
(self, x)
src/model.py:25
Method
forward
(self, x)
src/model.py:47
Method
forward
(self, x)
src/model.py:62
Method
forward
(self, x)
src/model.py:76
Method
forward
(self, lx, rx)
src/model.py:104
Method
forward
(self, x1, x2, x3)
src/model.py:152
Method
forward
(self, x)
src/model.py:175
Method
forward
(self, x)
src/model.py:184