Functions — github.com/ChrisChen1023/SEM-Net

CodeHub MCPcopy Create free account

hub / github.com/ChrisChen1023/SEM-Net / functions

Functions87 in github.com/ChrisChen1023/SEM-Net

⨍ Functions 87 ◇ Types & classes 24

↓ 14 callersMethodpostprocess

↓ 11 callersMethod__init__

(self, inp_channels=4, out_channels=3, dim=48,

src/networks.py:296

↓ 8 callersMethodcompute_gram

↓ 8 callersFunctioncreate_dir

src/utils.py:12

↓ 6 callersMethodcuda

↓ 6 callersMethodlog

↓ 5 callersFunctionPositionalEncoding

(d_model, max_len=5000)

src/models.py:227

↓ 5 callersMethodresize

(self, img, height, width, centerCrop=True)

src/dataset.py:164

↓ 5 callersFunctionspectral_norm

(module, mode=True)

src/networks.py:47

↓ 4 callersMethod__init__

src/loss.py:110

↓ 4 callersFunctionimsave

src/utils.py:53

↓ 4 callersMethodsave

↓ 3 callersMethodbackward

(self, gen_loss = None, dis_loss = None)

src/models.py:199

↓ 3 callersFunctionmain

r"""starts the model Args: mode (int): 1: train, 2: test, reads from config file if not specified

↓ 2 callersFunctioncreate_mask

(width, height, mask_width, mask_height, x=None, y=None)

src/utils.py:17

↓ 2 callersMethodload_flist

src/dataset.py:178

↓ 2 callersMethodload_name

src/dataset.py:40

↓ 2 callersFunctionstitch_images

(inputs, *outputs, img_per_row=2)

src/utils.py:25

↓ 2 callersMethodto_tensor

src/dataset.py:159

↓ 2 callersMethodupdate

Updates the progress bar. Arguments: current: Index of current step. values: List of tuples: `(name,

src/utils.py:98

↓ 1 callersMethod__init__

(self, name, config)

src/models.py:12

↓ 1 callersMethodadd

(self, n, values=None)

src/utils.py:219

↓ 1 callersFunctiongenerate_stroke_mask

(im_size, max_parts=15, maxVertex=25, maxLength=100, maxBrushWidth=24, maxAngle=360)

src/dataset.py:210

↓ 1 callersMethodinit_weights

initialize network's weights init_type: normal | xavier | kaiming | orthogonal https://github.com/junyanz/pytorch-CycleGAN-an

src/networks.py:19

↓ 1 callersMethodload

↓ 1 callersFunctionload_config

r"""loads model config Args: mode (int): 1: train, 2: test, reads from config file if not specified

↓ 1 callersMethodload_item

src/dataset.py:44

↓ 1 callersMethodload_mask

(self, img, index)

src/dataset.py:80

↓ 1 callersMethodmetric

(self, gt, pre)

↓ 1 callersFunctionnp_free_form_mask

(maxVertex, maxLength, maxBrushWidth, maxAngle, h, w)

src/dataset.py:218

↓ 1 callersMethodprint

src/config.py:20

↓ 1 callersMethodprocess

(self, images, masks)

src/models.py:108

↓ 1 callersMethodtest

↓ 1 callersFunctionto_3d

src/networks.py:99

↓ 1 callersFunctionto_4d

src/networks.py:103

↓ 1 callersMethodtrain

(self, outputs, is_real, is_disc=None)

src/metrics.py:15

Method__getattr__

src/config.py:11

Method__getitem__

src/dataset.py:35

r""" type = nsgan | lsgan | hinge

(self, weights=[1.0, 1.0, 1.0, 1.0, 1.0])

src/loss.py:233

(self, config_path)

src/config.py:5

(self, target, width=25, verbose=1, interval=0.05, stateful_metrics=None)

src/utils.py:74

src/models.py:63

(self, config, flist, mask_flist, augment=True, training=True)

src/dataset.py:18

(self, max_val)

src/metrics.py:6

src/networks.py:16

(self, in_channels, use_sigmoid=True, use_spectral_norm=True, init_weights=True)

src/networks.py:54

(self, normalized_shape)

src/networks.py:108

(self, normalized_shape)

src/networks.py:125

(self, dim, LayerNorm_type)

src/networks.py:144

(self, dim, ffn_expansion_factor, bias)

src/networks.py:159

(self, dim, num_heads, ffn_expansion_factor, bias, LayerNorm_type)

src/networks.py:218

(self, in_c=3, embed_dim=48, bias=False)

src/networks.py:252

src/networks.py:266

src/networks.py:283

(self, dim, d_state = 16, d_conv = 4, expand = 2)

src/networks.py:406

src/dataset.py:32

Functionabs_smooth

src/models.py:221

Methodbackward_joint

(self, gen_loss = None, dis_loss = None)

src/models.py:212

Methodcreate_iterator

(self, batch_size)

src/dataset.py:198

src/loss.py:186

src/loss.py:236

(self, images, masks)

src/models.py:175

src/networks.py:85

src/networks.py:119

src/networks.py:137

src/networks.py:151

(self, x, spatial)

src/networks.py:188

src/networks.py:237

src/networks.py:257

(self, x, mask)

src/networks.py:274

(self, x, mask)

src/networks.py:289

(self, inp_img, mask_whole, mask_half, mask_quarter,mask_tiny, pos1, pos2, pos3, pos4, pos1_dec)

src/networks.py:362

(self, x, pe, mask)

src/networks.py:427

Functionimage_transforms

src/dataset.py:241

(img, title='')

src/utils.py:45

Methodinit_func

src/networks.py:26

src/models.py:22

Methodload_edge

(self, img, index, mask)

src/dataset.py:127

src/models.py:49