MCPcopy Create free account

hub / github.com/NVIDIA/semantic-segmentation / functions

Functions433 in github.com/NVIDIA/semantic-segmentation

↓ 1 callersFunctionadjust_saturation
Adjust color saturation of an image. Args: img (PIL Image): PIL Image to be adjusted. saturation_factor (float): How much to adj
transforms/transforms.py:232
↓ 1 callersFunctionassert_and_infer_cfg
Call this function in your script after you have finished setting all cfg values that are necessary (e.g., merging a config from a file, merging
config.py:205
↓ 1 callersFunctionaugment_list
()
datasets/randaugment.py:179
↓ 1 callersFunctionbatch_cholesky_inverse
Args: matrix, 4-D tensor, [N, C, M, M]. matrix must be a symmetric positive define matrix.
loss/rmi_utils.py:110
↓ 1 callersFunctionbatch_inv_test
test for function batch_cholesky_inverse()
loss/rmi_utils.py:152
↓ 1 callersFunctionbatch_low_tri_inv
Batched inverse of lower triangular matrices Args: L : a lower triangular matrix Ref: https://www.pugetsystems.com/labs/hpc/PyTorch-for-Scient
loss/rmi_utils.py:120
↓ 1 callersFunctioncalc_err_mask
calculate class-specific error masks
utils/trnval_utils.py:58
↓ 1 callersFunctioncalc_err_mask_all
calculate class-agnostic error masks
utils/trnval_utils.py:71
↓ 1 callersFunctioncalc_tile_locations
Divide an image into tiles to help us cover classes that are spread out. tile_size: size of tile to distribute image_size: original image
datasets/uniform.py:67
↓ 1 callersFunctioncheck_termination
(epoch)
train.py:303
↓ 1 callersFunctioncities_cv_split
Find cities that correspond to a given split of the data. We split the data such that a given city belongs to either train or val, but never
datasets/cityscapes.py:41
↓ 1 callersFunctionclass_centroids_all
intermediate function to call pooled_class_centroid
datasets/uniform.py:189
↓ 1 callersFunctionclass_centroids_image
For one image, calculate centroids for all classes present in image. item: image, image_name tile_size: num_classes: id2trainid:
datasets/uniform.py:84
↓ 1 callersFunctioncoarse_cities
Find coarse cities
datasets/cityscapes.py:92
↓ 1 callersMethodcolorize_mask
Colorize the segmentation mask
datasets/base_loader.py:94
↓ 1 callersMethodcustom_nll
NLL Relaxed Loss Implementation
loss/utils.py:179
↓ 1 callersFunctiondb_eval_boundary
Compute mean,recall and decay from per-frame evaluation. Calculates precision/recall for boundaries between foreground_mask and gt_mask using morp
utils/f_boundary.py:110
↓ 1 callersMethoddetect_peaks
Takes an image and detect the peaks usingthe local maximum filter. Returns a boolean mask of the peaks (i.e. 1 when the pixel
transforms/joint_transforms.py:599
↓ 1 callersMethoddisable_coarse
Turn off using coarse images in training
datasets/cityscapes.py:169
↓ 1 callersMethoddo_transforms
Do transformations to image and mask :returns: image, mask
datasets/base_loader.py:120
↓ 1 callersMethoddump_images
(self, img_name, mask, centroid, class_id, img)
datasets/base_loader.py:102
↓ 1 callersFunctioneval_minibatch
Evaluate a single minibatch of images. * calculate metrics * dump images There are two primary multi-scale inference types:
utils/trnval_utils.py:82
↓ 1 callersFunctionfast_hist
(pred, gtruth, num_classes)
utils/misc.py:50
↓ 1 callersMethodfeatures
Forward Pass through the each layer of SE network
network/SEresnext.py:337
↓ 1 callersMethodfill_colormap
(self)
datasets/cityscapes.py:218
↓ 1 callersMethodfill_colormap_and_names
Mapillary code for color map and class names Outputs ------- self.trainid_to_name self.color_mapping
datasets/mapillary.py:85
↓ 1 callersMethodfind_images
Find image and segmentation mask files and return a list of tuples of them.
datasets/base_loader.py:74
↓ 1 callersFunctionfixed_padding
(inputs, kernel_size, dilation)
network/xception.py:15
↓ 1 callersMethodforward_simple
(self, input, target, return_raw_cost=False)
loss/utils.py:255
↓ 1 callersMethodgather
(self, outputs, output_device)
utils/my_data_parallel.py:201
↓ 1 callersFunctionget_loss
Get the criterion based on the loss function args: commandline arguments return: criterion, criterion_val
loss/utils.py:40
↓ 1 callersFunctionget_model
Fetch Network Function Pointer
network/__init__.py:45
↓ 1 callersMethodget_num_inchannels
(self)
network/hrnetv2.py:227
↓ 1 callersFunctionget_optimizer
Decide Optimizer (Adam or SGD)
loss/optimizer.py:43
↓ 1 callersMethodget_params
Get a randomized transform to be applied on image. Arguments are same as that of __init__. Returns: Transform which rand
transforms/transforms.py:317
↓ 1 callersMethodimage_in_crop
(self, target_w, target_h, w, h, img, mask)
transforms/joint_transforms.py:126
↓ 1 callersFunctioninit_attn
(m)
network/utils.py:332
↓ 1 callersMethodinit_weights
(self, pretrained=cfg.MODEL.HRNET_CHECKPOINT)
network/hrnetv2.py:451
↓ 1 callersFunctionlog_det_by_cholesky
Args: matrix: matrix must be a positive define matrix. shape [N, C, D, D]. Ref: https://github.com/tensorflow/tensorflow/blob/r1.13/tensorf
loss/rmi_utils.py:95
↓ 1 callersMethodlogits
AvgPool and Linear Layer
network/SEresnext.py:348
↓ 1 callersFunctionmain
Main Function
train.py:324
↓ 1 callersFunctionmetrics_per_image
Calculate tp, fp, fn for one image
utils/misc.py:476
↓ 1 callersMethodnscale_forward
Hierarchical attention, primarily used for getting best inference results. We use attention at multiple scales, giving prior
network/mscale2.py:55
↓ 1 callersMethodnscale_forward
Hierarchical attention, primarily used for getting best inference results. We use attention at multiple scales, giving prior
network/mscale.py:114
↓ 1 callersMethodnscale_forward
Hierarchical attention, primarily used for getting best inference results. We use attention at multiple scales, giving prior
network/ocrnet.py:185
↓ 1 callersMethodnscale_fused_forward
multi-scale evaluation for model with fused_aspp feature Evaluation must happen in two directions: from low to high to feed
network/mscale.py:88
↓ 1 callersFunctionold_make_attn_head
(in_ch, bot_ch, out_ch)
network/utils.py:370
↓ 1 callersMethodonly_coarse
Turn on using coarse images in training
datasets/cityscapes.py:175
↓ 1 callersMethodparallel_apply
(self, replicas, inputs, kwargs)
utils/my_data_parallel.py:198
↓ 1 callersFunctionpooled_class_centroids_all
Calculate class centroids for all classes for all images for all tiles. items: list of (image_fn, label_fn) tile size: size of tile r
datasets/uniform.py:138
↓ 1 callersFunctionprep_experiment
Make output directories, setup logging, Tensorboard, snapshot code.
utils/misc.py:70
↓ 1 callersFunctionprint_evaluate_results
If single scale: just print results for default scale else print all scale results Inputs: hist = histogram for defaul
utils/misc.py:420
↓ 1 callersMethodread_images
(self, img_path, mask_path, mask_out=False)
datasets/base_loader.py:152
↓ 1 callersMethodrecurse_fuse_fwd
recursive eval for n-scales target resolution is fixed at 1.0 [0.5, 1.0]: p_0.5, aspp_0.5, attn_0.5 = fwd(attn,
network/mscale.py:53
↓ 1 callersMethodreplicate
(self, module, device_ids)
utils/my_data_parallel.py:192
↓ 1 callersMethodreset
(self)
utils/misc.py:490
↓ 1 callersFunctionresnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
network/Resnet.py:233
↓ 1 callersFunctionresnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
network/Resnet.py:221
↓ 1 callersFunctionrestore_net
(net, checkpoint)
loss/optimizer.py:132
↓ 1 callersFunctionrestore_opt
(optimizer, checkpoint)
loss/optimizer.py:127
↓ 1 callersFunctionrestore_snapshot
Restore weights and optimizer (if needed ) for resuming job.
loss/optimizer.py:110
↓ 1 callersMethodrmi_lower_bound
calculate the lower bound of the region mutual information. Args: labels_4D : [N, C, H, W], dtype=float32
loss/rmi.py:139
↓ 1 callersMethodscatter
(self, inputs, kwargs, device_ids)
utils/my_data_parallel.py:195
↓ 1 callersFunctionse_resnext101_32x4d
Defination For SE Resnext101
network/SEresnext.py:396
↓ 1 callersFunctionse_resnext50_32x4d
Defination For SE Resnext50
network/SEresnext.py:383
↓ 1 callersMethodset_epoch
(self, epoch)
datasets/sampler.py:105
↓ 1 callersFunctiontensor_to_pil
(img)
utils/misc.py:88
↓ 1 callersFunctiontorch_version_float
()
config.py:193
↓ 1 callersFunctiontrain
Runs the training loop per epoch train_loader: Data loader for train net: thet network optimizer: optimizer curr_epoch: current e
train.py:465
↓ 1 callersMethodtwo_scale_forward
(self, inputs)
network/mscale2.py:129
↓ 1 callersMethodtwo_scale_forward
(self, inputs)
network/mscale.py:182
↓ 1 callersMethodtwo_scale_forward
Do we supervised both aux outputs, lo and high scale? Should attention be used to combine the aux output? Normally we only su
network/ocrnet.py:264
↓ 1 callersFunctionupdate_dataset_cfg
(num_classes, ignore_label)
config.py:371
↓ 1 callersFunctionupdate_dataset_inst
(dataset_inst)
config.py:379
↓ 1 callersFunctionupdate_epoch
(epoch)
config.py:364
↓ 1 callersFunctionvalidate_topn
Find worse case failures ... Only single GPU for now First pass = calculate TP, FP, FN pixels per image per class Take these stat
utils/trnval_utils.py:201
↓ 1 callersMethodwrite_summaries
write out tensorboard write out html webpage summary only update tensorboard if was a best epoch always update webpa
utils/misc.py:388
FunctionAutoContrast
(pair, _)
datasets/randaugment.py:81
FunctionBrightness
(pair, v)
datasets/randaugment.py:133
FunctionColor
(pair, v)
datasets/randaugment.py:127
FunctionContrast
(pair, v)
datasets/randaugment.py:121
FunctionCutout
(pair, v)
datasets/randaugment.py:145
FunctionDeepV3EffB4
(num_classes, criterion)
network/mscale.py:349
FunctionDeepV3EffB4Fuse
(num_classes, criterion)
network/mscale.py:354
FunctionDeepV3PlusEffB4
(num_classes, criterion)
network/deepv3.py:121
FunctionDeepV3PlusR50
(num_classes, criterion)
network/deepv3.py:100
FunctionDeepV3PlusSRNX101
(num_classes, criterion)
network/deepv3.py:104
FunctionDeepV3PlusSRNX50
(num_classes, criterion)
network/deepv3.py:96
FunctionDeepV3PlusW38
(num_classes, criterion)
network/deepv3.py:108
FunctionDeepV3PlusW38I
(num_classes, criterion)
network/deepv3.py:112
FunctionDeepV3PlusX71
(num_classes, criterion)
network/deepv3.py:117
FunctionDeepV3R50
(num_classes, criterion)
network/mscale2.py:227
FunctionDeepV3R50
(num_classes, criterion)
network/mscale.py:331
FunctionDeepV3R50
(num_classes, criterion)
network/attnscale.py:185
FunctionDeepV3R50
(num_classes, criterion)
network/deepv3.py:164
FunctionDeepV3R50B
(num_classes, criterion)
network/attnscale.py:190
FunctionDeepV3R50BP
(num_classes, criterion)
network/attnscale.py:370
FunctionDeepV3W38
(num_classes, criterion)
network/mscale.py:335
FunctionDeepV3W38
(num_classes, criterion)
network/attnscale.py:195
FunctionDeepV3W38Fuse
(num_classes, criterion)
network/mscale.py:339
← previousnext →101–200 of 433, ranked by callers