Functions — github.com/chenfengxu714/SqueezeSegV3

hub / github.com/chenfengxu714/SqueezeSegV3 / functions

Functions132 in github.com/chenfengxu714/SqueezeSegV3

⨍ Functions 132◇ Types & classes 32

↓ 15 callersMethodsize

Return the size of the point cloud.

src/common/laserscan.py:51

↓ 13 callersMethodupdate

(self, val, n=1)

src/common/avgmeter.py:16

↓ 7 callersMethodget_n_classes

(self)

src/tasks/semantic/dataset/kitti/parser.py:341

↓ 7 callersMethodopen_label

Open raw scan and fill in attributes

src/common/laserscan.py:229

↓ 7 callersMethodopen_scan

Open raw scan and fill in attributes

src/common/laserscan.py:58

↓ 6 callersMethodmap

(label, mapdict)

src/tasks/semantic/dataset/kitti/parser.py:187

↓ 6 callersMethodreset

(self)

src/tasks/semantic/modules/ioueval.py:27

↓ 5 callersMethod_make_dec_layer

(self, block, planes, bn_d=0.1, stride=2,flag=True)

src/tasks/semantic/decoders/SAC.py:91

↓ 5 callersMethod_make_enc_layer

(self, block, planes, blocks, stride, DS, bn_d=0.1)

src/backbones/SAC.py:129

↓ 5 callersFunction_unsqueeze_ft

add new dementions at the front and the tail

src/common/sync_batchnorm/batchnorm.py:31

↓ 5 callersMethodaddBatch

(self, x, y)

src/tasks/semantic/modules/ioueval.py:33

↓ 5 callersMethodgetIoU

(self)

src/tasks/semantic/modules/ioueval.py:69

↓ 5 callersMethodgetacc

(self)

src/tasks/semantic/modules/ioueval.py:77

↓ 5 callersMethodrun_layer

(self, xyz, feature, layer, skips, os, flag=True)

src/backbones/SAC.py:146

↓ 5 callersMethodrun_layer

(self, x, layer, skips, os)

src/tasks/semantic/decoders/SAC.py:110

↓ 4 callersMethodreset

Reset scan members.

src/common/laserscan.py:18

↓ 3 callersMethodget

(self)

src/common/sync_batchnorm/comm.py:32

↓ 3 callersMethodinfer_subset

(self, loader, to_orig_fn)

src/tasks/semantic/modules/user.py:89

↓ 3 callersMethodput

(self, result)

src/common/sync_batchnorm/comm.py:26

↓ 3 callersMethodupdate_scan

(self)

src/common/laserscanvis.py:127

↓ 2 callersMethod_check_input_dim

(self, input)

src/common/sync_batchnorm/batchnorm.py:193

↓ 2 callersFunction_sum_ft

sum over the first and last dimention

src/common/sync_batchnorm/batchnorm.py:26

↓ 2 callersFunctionexecute_replication_callbacks

Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked with

src/common/sync_batchnorm/replicate.py:27

↓ 2 callersMethodgetStats

(self)

src/tasks/semantic/modules/ioueval.py:57

↓ 2 callersMethodget_mpl_colormap

(cmap_name)

src/tasks/semantic/modules/trainer.py:162

↓ 2 callersMethodget_test_set

(self)

src/tasks/semantic/dataset/kitti/parser.py:329

↓ 2 callersMethodget_train_set

(self)

src/tasks/semantic/dataset/kitti/parser.py:315

↓ 2 callersMethodget_valid_set

(self)

src/tasks/semantic/dataset/kitti/parser.py:322

↓ 2 callersMethodhisto_summary

Log a histogram of the tensor of values.

src/common/logger.py:50

↓ 2 callersMethodinfer

(self)

src/tasks/semantic/modules/user.py:74

↓ 2 callersMethodmake_log_img

(depth, mask, pred, gt, color_fn)

src/tasks/semantic/modules/trainer.py:171

↓ 2 callersMethodsave_checkpoint

(self, logdir, suffix="")

src/tasks/semantic/modules/segmentator.py:228

↓ 2 callersMethodstep

(self, epoch=None)

src/common/warmupLR.py:43

↓ 2 callersMethodtrain

(self)

src/tasks/semantic/modules/trainer.py:210

↓ 1 callersMethod__data_parallel_replicate__

(self, ctx, copy_id)

src/common/sync_batchnorm/batchnorm.py:87

↓ 1 callersMethod__init__

(self, inplanes, expand1x1_planes, bn_d = 0.1)

src/backbones/SAC.py:11

↓ 1 callersMethod__init__

(self, n_classes, device, ignore=None)

src/tasks/semantic/modules/ioueval.py:13

↓ 1 callersMethod__init__

(self, params, nclasses)

src/tasks/semantic/postproc/CRF.py:79

↓ 1 callersMethod__init__

(self, params, stub_skips, OS=32, feature_depth=1024)

src/tasks/semantic/decoders/SAC.py:43

↓ 1 callersMethod__init__

(self, project=False, H=64, W=2048, fov_up=3.0, fov_down=-25.0)

src/common/laserscan.py:10

↓ 1 callersMethod__init__

Args: master_callback: a callback to be invoked after having collected messages from slave devices.

src/common/sync_batchnorm/comm.py:68

↓ 1 callersMethod_compute_mean_std

Compute the mean and standard-deviation with sum and square-sum. This method also maintains the moving average on the master device.

src/common/sync_batchnorm/batchnorm.py:120

↓ 1 callersMethodaddBorderBatch1d

range_y=target as img, x=preds, y=targets, px,py=idxs of points of pointcloud in range img WARNING: Only batch size 1 works for now

src/tasks/semantic/modules/ioueval.py:108

↓ 1 callersMethodcolorize

Colorize pointcloud with the color of each semantic label

src/common/laserscan.py:270

↓ 1 callersFunctionconvert_model

Traverse the input module and its child recursively and replace all instance of torch.nn.modules.batchnorm.BatchNorm*N*d to SynchronizedBatc

src/common/sync_batchnorm/batchnorm.py:327

↓ 1 callersMethoddestroy

(self)

src/common/laserscanvis.py:212

↓ 1 callersMethoddo_label_projection

(self)

src/common/laserscan.py:279

↓ 1 callersMethoddo_range_projection

Project a pointcloud into a spherical projection image.projection. Function takes no arguments because it can be also called externally

src/common/laserscan.py:107

↓ 1 callersFunctionget_gaussian_kernel

(kernel_size=3, sigma=2, channels=1)

src/tasks/semantic/postproc/KNN.py:11

↓ 1 callersMethodget_input_depth

(self)

src/backbones/SAC.py:177

↓ 1 callersMethodget_last_depth

(self)

src/tasks/semantic/decoders/SAC.py:132

↓ 1 callersMethodget_mpl_colormap

(self, cmap_name)

src/common/laserscanvis.py:116

↓ 1 callersMethodget_train_size

(self)

src/tasks/semantic/dataset/kitti/parser.py:332

↓ 1 callersMethodinfer_subset

(self, loader, to_orig_fn)

src/tasks/semantic/modules/demo_user.py:84

↓ 1 callersFunctionis_label

(filename)

src/tasks/semantic/dataset/kitti/parser.py:15

↓ 1 callersFunctionis_scan

(filename)

src/tasks/semantic/dataset/kitti/parser.py:11

↓ 1 callersMethodonehot1dspatial

(self, x)

src/common/onehot.py:15

↓ 1 callersMethodonehot2dspatial

(self, x)

src/common/onehot.py:40

↓ 1 callersMethodregister_slave

Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` object which can be

src/common/sync_batchnorm/comm.py:85

↓ 1 callersMethodreset

(self)

src/tasks/semantic/modules/ioueval.py:104

↓ 1 callersMethodreset

Reset scan members.

src/common/laserscan.py:205

↓ 1 callersMethodreset

(self)

src/common/avgmeter.py:10

↓ 1 callersMethodreset

Reset.

src/common/laserscanvis.py:30

↓ 1 callersMethodrun

(self)

src/common/laserscanvis.py:218

↓ 1 callersMethodrun_master

Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master device), and

src/common/sync_batchnorm/comm.py:103

↓ 1 callersMethodrun_slave

(self, msg)

src/common/sync_batchnorm/comm.py:50

↓ 1 callersMethodsave_to_log

(logdir, logger, info, epoch, w_summary=False, model=None, img_summary=False, imgs=[])

src/tasks/semantic/modules/trainer.py:188

↓ 1 callersMethodscalar_summary

Log a scalar variable.

src/common/logger.py:18

↓ 1 callersMethodset_label

Set points for label not from file but from np

src/common/laserscan.py:248

↓ 1 callersMethodset_points

Set scan attributes (instead of opening from file)

src/common/laserscan.py:82

↓ 1 callersMethodto_color

(self, label)

src/tasks/semantic/dataset/kitti/parser.py:358

↓ 1 callersMethodto_xentropy

(self, label)

src/tasks/semantic/dataset/kitti/parser.py:354

↓ 1 callersMethodtrain_epoch

(self, train_loader, model, criterion, optimizer, epoch, evaluator, scheduler, color_fn, report=10, show_scans

src/tasks/semantic/modules/trainer.py:295

↓ 1 callersMethodvalidate

(self, val_loader, model, criterion, evaluator, class_func, color_fn, save_scans)

src/tasks/semantic/modules/trainer.py:402

Method__getitem__

(self, index)

src/tasks/semantic/dataset/kitti/parser.py:110

Method__getstate__

(self)

src/common/sync_batchnorm/comm.py:79

Method__init__

(self, params)

src/backbones/SAC.py:59

Method__init__

(self, ARCH, DATA, datadir, logdir, modeldir)

src/tasks/semantic/modules/user.py:24

Method__init__

(self, ARCH, nclasses, path=None, path_append="", strict=False)

src/tasks/semantic/modules/segmentator.py:12

Method__init__

(self, ARCH, DATA, datadir, logdir, path=None)

src/tasks/semantic/modules/trainer.py:30

Method__init__

(self, ARCH, DATA, datadir, logdir, modeldir)

src/tasks/semantic/modules/demo_user.py:25

Method__init__

(self, n_classes, device, ignore=None, border_size=1, kern_conn=4)

src/tasks/semantic/modules/ioueval.py:86

Method__init__

(self, root, # directory where data is sequences, # sequences for this data (e.g. [1,3,4

src/tasks/semantic/dataset/kitti/parser.py:21

Method__init__

(self, root, # directory for data train_sequences, # sequences to

src/tasks/semantic/dataset/kitti/parser.py:215

Method__init__

(self, params, nclasses)

src/tasks/semantic/postproc/KNN.py:37

Method__init__

Get the binary border mask of a labeled 2d range image. Args: nclasses(int) : The number of classes labeled in the input image

src/tasks/semantic/postproc/borderMask.py:93

Method__init__

(self, h, w, sigma, nclasses)

src/tasks/semantic/postproc/CRF.py:13

Method__init__

(self, inplanes, planes, bn_d=0.1)

src/tasks/semantic/decoders/SAC.py:10

Method__init__

(self, sem_color_dict=None, project=False, H=64, W=2048, fov_up=3.0, fov_down=-25.0, max_classes=300)

src/common/laserscan.py:174

Method__init__

Create a summary writer logging to log_dir.

src/common/logger.py:14

Method__init__

(self)

src/common/avgmeter.py:7

Method__init__

(self, scan, scan_names, label_names, offset=0, semantics=True, instances=False)

src/common/laserscanvis.py:14

Method__init__

(self, optimizer, lr, warmup_steps, momentum, decay)

src/common/warmupLR.py:13

Method__init__

(self, device, nclasses, spatial_dim=2)

src/common/onehot.py:9

Method__init__

(self)

src/common/sync_batchnorm/comm.py:21

Method__init__

(self, num_features, eps=1e-5, momentum=0.1, affine=True)

src/common/sync_batchnorm/batchnorm.py:42

Method__len__

(self)

src/tasks/semantic/dataset/kitti/parser.py:183

Method__len__

(self)

src/common/laserscan.py:55

Method__setstate__

(self, state)

src/common/sync_batchnorm/comm.py:82

Method_check_input_dim

(self, input)

src/common/sync_batchnorm/batchnorm.py:256

next →1–100 of 132, ranked by callers