Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Haochen-Wang409/U2PL
/ types & classes
Types & classes
33 in github.com/Haochen-Wang409/U2PL
⨍
Functions
166
◇
Types & classes
33
↓ 20 callers
Class
AverageMeter
Computes and stores the average and current value
u2pl/utils/utils.py:438
↓ 5 callers
Class
ModelBuilder
u2pl/models/model_helper.py:9
↓ 5 callers
Class
ResNet
u2pl/models/resnet.py:143
↓ 3 callers
Class
city_dset
u2pl/dataset/cityscapes.py:18
↓ 3 callers
Class
voc_dset
u2pl/dataset/pascal_voc.py:17
↓ 2 callers
Class
ASPP
Reference: Chen, Liang-Chieh, et al. *"Rethinking Atrous Convolution for Semantic Image Segmentation."*
u2pl/models/base.py:11
↓ 2 callers
Class
OhemCrossEntropy2dTensor
Ohem Cross Entropy Tensor Version
u2pl/utils/loss_helper.py:451
↓ 1 callers
Class
Aux_Module
u2pl/models/decoder.py:127
↓ 1 callers
Class
Criterion
u2pl/utils/loss_helper.py:258
↓ 1 callers
Class
CriterionOhem
u2pl/utils/loss_helper.py:323
↓ 1 callers
Class
GaussianBlur
u2pl/dataset/augmentation.py:325
↓ 1 callers
Class
LRScheduler
u2pl/utils/lr_helper.py:42
Class
BaseDataset
u2pl/dataset/base.py:7
Class
BasicBlock
u2pl/models/resnet.py:44
Class
Bottleneck
u2pl/models/resnet.py:93
Class
Compose
Composes several segsegtransforms together. Args: segtransforms (List[Transform]): list of segtransforms to compose. Example:
u2pl/dataset/augmentation.py:15
Class
Crop
Crops the given tensor. Args: size (sequence or int): Desired output size of the crop. If size is an int instead of sequence like
u2pl/dataset/augmentation.py:214
Class
Cutmix
Randomly mask out one or more patches from an image. Args: n_holes (int): Number of patches to cut out of each image. length (int)
u2pl/dataset/augmentation.py:398
Class
Cutout
Randomly mask out one or more patches from an image. Args: n_holes (int): Number of patches to cut out of each image. length (int)
u2pl/dataset/augmentation.py:349
Class
DistributedGivenIterationSampler
u2pl/utils/dist_helper.py:56
Class
DistributedGivenIterationSampler
u2pl/dataset/sampler.py:12
Class
Normalize
Given mean and std of each channel Will normalize each channel of the torch.*Tensor (C*H*W), i.e. channel = (channel - mean) / std
u2pl/dataset/augmentation.py:89
Class
OhemCrossEntropy2d
u2pl/utils/loss_helper.py:363
Class
RandResize
Randomly resize image & label with scale factor in [scale_min, scale_max]
u2pl/dataset/augmentation.py:159
Class
RandRotate
Randomly rotate image & label with rotate factor in [rotate_min, rotate_max]
u2pl/dataset/augmentation.py:269
Class
RandomGaussianBlur
u2pl/dataset/augmentation.py:315
Class
RandomHorizontalFlip
u2pl/dataset/augmentation.py:299
Class
RandomVerticalFlip
u2pl/dataset/augmentation.py:307
Class
Resize
Resize the input tensor to the given size. 'size' is a 2-element tuple or list in the order of (h, w)
u2pl/dataset/augmentation.py:114
Class
ResizeLongSize
Resize the long size of the input image into fix size
u2pl/dataset/augmentation.py:132
Class
ToTensor
u2pl/dataset/augmentation.py:50
Class
dec_deeplabv3
u2pl/models/decoder.py:8
Class
dec_deeplabv3_plus
u2pl/models/decoder.py:45