Functions — github.com/EryiXie/PlaneRecNet

CodeHub MCPcopy Create free account

hub / github.com/EryiXie/PlaneRecNet / functions

Functions218 in github.com/EryiXie/PlaneRecNet

⨍ Functions 218◇ Types & classes 49

↓ 1 callersMethodset_val

(self, new_val)

utils/utils.py:65

↓ 1 callersFunctionsetup_eval

↓ 1 callersMethodsplit_feats

planerecnet.py:114

↓ 1 callersFunctiontensorborad_visual_log

(net: PlaneRecNet, dataset, writer: SummaryWriter, iteration, eval_nums)

↓ 1 callersFunctiontotal_time

Returns the total amount accumulated across all functions in seconds.

utils/timer.py:134

↓ 1 callersFunctiontrain

(self, img, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:28

(self, image, depth, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:46

(self, image, depth, masks, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:118

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:148

(self, image, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:191

(self, image, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:203

(self, image, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:216

(self, image, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:234

(self, image, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:247

(self, tensor, depth, masks=None, boxes=None, labels=None)

data/augmentations.py:255

(self, cvimage, depth, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:260

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:265

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:279

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:293

Args: image (Tensor): image tensor to be transformed Return: a tensor with channels swapped according to swap

data/augmentations.py:321

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:348

(self, img, depth, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:377

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:406

(self, image, depth, masks, boxes, labels, plane_paras)

data/augmentations.py:433

(self, img, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:480

(self, img, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:492

Method__enter__

utils/timer.py:157

(self, e, ev, t)

utils/timer.py:160

Method__getitem__

Args: index (int): Index Returns: tuple: Tuple (image, instances, depth).

data/datasets.py:42

SOLOv2 Instance Head.

planerecnet.py:293

SOLOv2 Mask Head.

planerecnet.py:395

planerecnet.py:501

(self, net:PlaneRecNet, criterion:PlaneRecNetLoss)

(self, max_window_size=1000)

utils/utils.py:15

(self, length, max_val)

utils/utils.py:57

(self, model_name:str, epoch:int, iteration:int)

utils/utils.py:109

(self, fn_name, use_stack=True)

utils/timer.py:153

(self, config_dict)

data/config.py:51

(self, transforms)

data/augmentations.py:25

(self, resize_gt=True, mean=MEANS, pad_gt=True)

data/augmentations.py:40

(self, resize_gt=True)

data/augmentations.py:144

(self, lower=0.5, upper=1.5)

data/augmentations.py:185

(self, delta=18.0)

data/augmentations.py:199

(self, current='BGR', transform='HSV')

data/augmentations.py:212

(self, lower=0.5, upper=1.5)

data/augmentations.py:227

(self, delta=32)

data/augmentations.py:242

data/augmentations.py:318

data/augmentations.py:337

(self, transform, mean, std, in_channel_order)

data/augmentations.py:368

(self, lower_degree=3, upper_degree=12, angle=180)

data/augmentations.py:398

(self, mean=0, var=0.0002)

data/augmentations.py:429

(self, mean=MEANS, std=STD)

data/augmentations.py:467

(self, mean=MEANS, std=STD)

data/augmentations.py:486

data/augmentations.py:503

(self, image_path, anno_file, transform=None, dataset_name=None, has_gt=True, has_pos=True)

data/datasets.py:26

(self, image_path, anno_file, transform=None, dataset_name=None, has_gt=True, has_pos=True)

data/datasets.py:183

(self, image_path, anno_file, transform=None, dataset_name=None, has_gt=True, has_pos=True)

data/datasets.py:225

(self, in_channels, out_channels, kernel_size=3,

models/dcn.py:12

( self, in_channels, start_level=0 )

models/fpn.py:7

( self, layers, dcn_layers=[0, 0, 0, 0], dcn_interval=1, atrous_layers

models/backbone.py:79

(self, input_size, delta_cos=0.867, delta_diff_x=0.01, delta_diff_y=0.01, de

models/functions/vnl.py:173

models/functions/losses.py:20

(self, alpha: float = -1, gamma: float = 2, reduction: str = "none")

models/functions/losses.py:332

models/functions/losses.py:356

(self, clamp_val=1e-9, reduction: str = "none")

models/functions/losses.py:372

utils/utils.py:50

utils/utils.py:47

utils/utils.py:85

data/datasets.py:160

utils/utils.py:44

utils/utils.py:88

Functiondetection_collate

Custom collate fn for dealing with batches of images that have a different number of associated object annotations (bounding boxes). Argument

data/datasets.py:250

Functiondisable

Disables the given function name fom being considered for the average or outputted in print_stats.

utils/timer.py:35

Functiondisable_all

utils/timer.py:25

Functiondo_nothing

(img=None, depth=None, masks=None, boxes=None, labels=None, plane_paras=None)

data/augmentations.py:456

Enables function names disabled by disable.

utils/timer.py:40

Functionenable_all

utils/timer.py:30

planerecnet.py:73

Arguments: features (list[Tensor]): FPN feature map tensors in high to low resolution. Each tensor in the list co

planerecnet.py:355

Arguments: features (list[Tensor]): FPN feature map tensors in high to low resolution. Each tensor in the list co

planerecnet.py:467

(self, feature_maps, seg_preds, kernel_preds)

planerecnet.py:586

Args: - batched_images: Tensor, each images in (C, H, W) format. - batched_gt_instances: Dict of Tensor, ground truth

data/augmentations.py:510

models/dcn.py:52

models/fpn.py:45

models/backbone.py:53

Returns a list of convouts for each layer.

models/backbone.py:197

(self, pred_depth, gt_masks, gt_planes, gt_depth, k_maritix, select=True)

models/functions/vnl.py:119

Virtual normal loss. :param pred_depth: predicted depth map, [B,W,H,C] :param data: target label, ground truth depth, [B, W,

models/functions/vnl.py:312

models/functions/losses.py:53

(self, seg_masks, gradient_map)

models/functions/losses.py:297

(self, input, target)

models/functions/losses.py:338

(self, input, target)

models/functions/losses.py:360

(self, input, target, valid_mask)

models/functions/losses.py:377

(self, outputs, output_device)

Methodget_camera_matrix

(self, rgb_file_name)

data/datasets.py:192

Methodget_camera_matrix

(self, rgb_file_name)

data/datasets.py:236

Methodget_depth_path

(self, rgb_file_name)

data/datasets.py:187

← previous next →101–200 of 218, ranked by callers