Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Seung-Hun-Lee/DRANet
/ functions
Functions
117 in github.com/Seung-Hun-Lee/DRANet
⨍
Functions
117
◇
Types & classes
21
↓ 8 callers
Method
__init__
(self)
models/DRANet.py:130
↓ 8 callers
Method
_make_layer
(self, block, planes, blocks, stride=1, dilation=1, new_level=True, residual=True)
models/drn.py:181
↓ 4 callers
Function
cadt
(source, target, style, W=10)
models/cadt.py:27
↓ 4 callers
Method
train
(self)
trainer.py:527
↓ 3 callers
Method
_img_transform
(self, image)
dataloader/Cityscapes.py:126
↓ 3 callers
Function
check_integrity
(fpath: str, md5: Optional[str] = None)
dataloader/utils.py:37
↓ 3 callers
Method
eval
(self, cv)
trainer.py:438
↓ 3 callers
Function
init_params
Init layer parameters.
utils.py:37
↓ 3 callers
Method
set_train
(self)
trainer.py:231
↓ 3 callers
Method
set_zero_grad
(self)
trainer.py:220
↓ 3 callers
Function
slice_patches
(imgs, hight_slice=2, width_slice=4)
utils.py:137
↓ 2 callers
Function
MIOU
(GT, Pred, num_class=19)
miou.py:4
↓ 2 callers
Method
__init__
(self, block, layers, num_cls=19, channels=(16, 32, 64, 128, 256, 512, 512, 512),
models/drn.py:113
↓ 2 callers
Method
_mask_transform
(self, gt_image)
dataloader/Cityscapes.py:142
↓ 2 callers
Method
_train_sync_transform
:param image: PIL input image :param gt_image: PIL input gt_image :return:
dataloader/Cityscapes.py:98
↓ 2 callers
Method
_val_sync_transform
(self, img, mask)
dataloader/Cityscapes.py:117
↓ 2 callers
Function
conv3x3
(in_planes, out_planes, stride=1, padding=1, dilation=1)
models/drn.py:29
↓ 2 callers
Function
decode_labels
Decode batch of segmentation masks. Args: mask: result of inference after taking argmax. num_images: number of images to decode from
dataloader/Cityscapes.py:153
↓ 2 callers
Function
format_time
(seconds)
utils.py:104
↓ 2 callers
Function
gen_bar_updater
()
dataloader/utils.py:13
↓ 2 callers
Function
get_args
()
args.py:16
↓ 2 callers
Function
gram
(x)
models/cadt.py:5
↓ 2 callers
Function
imshow
(img)
dataloader/MNIST_M.py:50
↓ 2 callers
Function
progress_bar
(current, total, msg=None)
utils.py:60
↓ 2 callers
Function
safe_load_state_dict
Copies parameters and buffers from :attr:`state_dict` into this module and its descendants. Any params in :attr:`state_dict` that do not match
models/utils.py:10
↓ 2 callers
Method
save_networks
(self)
trainer.py:116
↓ 2 callers
Method
set_default
(self)
trainer.py:97
↓ 2 callers
Method
set_eval
(self)
trainer.py:242
↓ 2 callers
Method
set_networks
(self)
trainer.py:146
↓ 1 callers
Method
_check_input_dim
(self, input)
models/batchinstancenorm.py:43
↓ 1 callers
Function
_get_confirm_token
(response: "requests.models.Response")
dataloader/utils.py:188
↓ 1 callers
Function
_get_redirect_url
(url: str, max_hops: int = 10)
dataloader/utils.py:45
↓ 1 callers
Function
_is_gzip
(filename: str)
dataloader/utils.py:226
↓ 1 callers
Function
_is_tar
(filename: str)
dataloader/utils.py:214
↓ 1 callers
Function
_is_targz
(filename: str)
dataloader/utils.py:218
↓ 1 callers
Function
_is_tarxz
(filename: str)
dataloader/utils.py:210
↓ 1 callers
Function
_is_tgz
(filename: str)
dataloader/utils.py:222
↓ 1 callers
Function
_is_zip
(filename: str)
dataloader/utils.py:230
↓ 1 callers
Function
_quota_exceeded
(response: "requests.models.Response")
dataloader/utils.py:140
↓ 1 callers
Function
_save_response_content
( response: "requests.models.Response", destination: str, chunk_size: int = 32768, # type: ignore[name-de
dataloader/utils.py:196
↓ 1 callers
Function
cadt_gram
(gram, con_sim)
models/cadt.py:43
↓ 1 callers
Function
calculate_md5
(fpath: str, chunk_size: int = 1024 * 1024)
dataloader/utils.py:25
↓ 1 callers
Function
check_dirs
(dirs)
args.py:6
↓ 1 callers
Function
check_md5
(fpath: str, md5: str, **kwargs: Any)
dataloader/utils.py:33
↓ 1 callers
Method
consistency
(self, contents, styles, contents_converted, styles_converted, converts)
loss_functions.py:96
↓ 1 callers
Method
content_perceptual
(self, perceptual, perceptual_converted)
loss_functions.py:81
↓ 1 callers
Method
dis
(self, real, fake)
loss_functions.py:55
↓ 1 callers
Function
download_url
Download a file from a url and place it in root. Args: url (str): URL to download file from root (str): Directory to place downloa
dataloader/utils.py:59
↓ 1 callers
Function
drn26
(pretrained=True, finetune=True, out_map=True, **kwargs)
models/drn.py:254
↓ 1 callers
Function
extract_archive
(from_path: str, to_path: Optional[str] = None, remove_finished: bool = False)
dataloader/utils.py:234
↓ 1 callers
Method
gen
(self, fake)
loss_functions.py:71
↓ 1 callers
Method
get_batch
(self, batch_data_iter)
trainer.py:246
↓ 1 callers
Function
get_dataset
(dataset, batch, imsize, workers)
dataset.py:10
↓ 1 callers
Method
id2trainId
(self, label, reverse=False, ignore_label=-1)
dataloader/Cityscapes.py:78
↓ 1 callers
Function
iterable_to_str
(iterable: Iterable)
dataloader/utils.py:282
↓ 1 callers
Method
load_networks
(self, step)
trainer.py:131
↓ 1 callers
Function
loss_weights
You must set these hyperparameters to apply our method to other datasets. These hyperparameters may not be the optimal value for your
loss_functions.py:5
↓ 1 callers
Method
print_loss
(self)
trainer.py:506
↓ 1 callers
Method
recon
(self, imgs, recon_imgs)
loss_functions.py:49
↓ 1 callers
Function
set_converts
train_converts: converts for image-to-image translation training test_converts: converts for domain adaptation test tensorboard_converts:
trainer.py:19
↓ 1 callers
Method
set_optimizers
(self)
trainer.py:192
↓ 1 callers
Method
style_perceptual
(self, style_gram, style_gram_converted)
loss_functions.py:88
↓ 1 callers
Method
task
(self, pred, gt)
loss_functions.py:104
↓ 1 callers
Method
tensor_board_log
(self, imgs, labels)
trainer.py:377
↓ 1 callers
Method
test
(self)
trainer.py:567
↓ 1 callers
Method
train_dis
(self, imgs)
trainer.py:256
↓ 1 callers
Method
train_esg
(self, imgs)
trainer.py:330
↓ 1 callers
Method
train_task
(self, imgs, labels)
trainer.py:291
Function
MIOU_score
(GT, Pred, num_class=19)
miou.py:13
Method
__getitem__
(self, item)
dataloader/Cityscapes.py:84
Method
__getitem__
(self, idx)
dataloader/MNIST_M.py:30
Method
__getitem__
(self, item)
dataloader/GTA5.py:48
Method
__init__
(self, args)
trainer.py:62
Method
__init__
(self, args)
loss_functions.py:45
Method
__init__
(self, list_path='./data_list/Cityscapes', split='train', c
dataloader/Cityscapes.py:44
Method
__init__
(self, root, train, transform=None)
dataloader/MNIST_M.py:13
Method
__init__
(self, list_path='./data_list/GTA5', split='train', crop_si
dataloader/GTA5.py:17
Method
__init__
(self, in_channels, filters=64, kernel_size=3, stride=1, padding=1)
models/DRANet.py:15
Method
__init__
(self, channels=3)
models/DRANet.py:39
Method
__init__
(self, imsize, converts, ch=64, down_scale=2)
models/DRANet.py:58
Method
__init__
(self, channels=512)
models/DRANet.py:88
Method
__init__
(self, channels=3, num_classes=10)
models/DRANet.py:104
Method
__init__
(self, channels=3)
models/DRANet.py:169
Method
__init__
(self, channels=3)
models/DRANet.py:194
Method
__init__
(self, channels=3)
models/DRANet.py:225
Method
__init__
(self, inplanes, planes, stride=1, downsample=None, dilation=(1, 1), residual=True)
models/drn.py:37
Method
__init__
(self, inplanes, planes, stride=1, downsample=None, dilation=(1, 1), residual=True)
models/drn.py:73
Method
__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True)
models/batchinstancenorm.py:8
Method
__len__
(self)
dataloader/Cityscapes.py:149
Method
__len__
(self)
dataloader/MNIST_M.py:27
Method
_check_input_dim
(self, input)
models/batchinstancenorm.py:49
Method
_check_input_dim
(self, input)
models/batchinstancenorm.py:55
Function
bar_update
(count, block_size, total_size)
dataloader/utils.py:16
Function
decorator
(cls)
models/utils.py:5
Function
download_and_extract_archive
( url: str, download_root: str, extract_root: Optional[str] = None, filename: Optional[str] =
dataloader/utils.py:261
Function
download_file_from_google_drive
Download a Google Drive file from and place it in root. Args: file_id (str): id of file to be downloaded root (str): Directory to
dataloader/utils.py:146
Function
drn42
(pretrained=False, finetune=False, out_map=True, **kwargs)
models/drn.py:269
Function
drn58
(pretrained=False, **kwargs)
models/drn.py:277
Method
forward
(self, x)
models/DRANet.py:32
Method
forward
(self, x)
models/DRANet.py:52
next →
1–100 of 117, ranked by callers