MCPcopy Create free account

hub / github.com/JudyYe/CVP / functions

Functions349 in github.com/JudyYe/CVP

↓ 14 callersFunctionadd_loss
register loss to total_loss and loss_dict
cvp/losses.py:98
↓ 11 callersMethod__init__
(self, layers)
cvp/pok_model.py:172
↓ 10 callersFunctionrefine_module
(din, dout, norm, act, num=2)
cvp/decoder.py:15
↓ 8 callersFunction_boxes_to_grid
Input: - boxes: FloatTensor of shape (O, 4) giving boxes in the [x0, y0, x1, y1] format in the [0, 1] coordinate space - H, W: Scal
cvp/layout.py:626
↓ 8 callersFunctionvid_batch_to_cuda
(batch)
utils/model_utils.py:10
↓ 7 callersMethodsave
(self,file='index')
PerceptualSimilarity/util/html.py:47
↓ 7 callersMethodsave_file_name
(self, pref, index, suff='', ext='.jpg')
cvp/evaluator.py:111
↓ 6 callersMethodadd_images
(self, iteration, images, name='')
cvp/logger.py:49
↓ 6 callersFunctionget_activation
(name)
cvp/layers.py:33
↓ 6 callersFunctionget_normalization_2d
(channels, normalization)
cvp/layers.py:22
↓ 5 callersMethod__init__
(self, input_dim, output_dim=None, edge_dim=None, pooling='avg', preact_normalization='batch'
cvp/graph.py:108
↓ 5 callersMethod__init__
under construction!!!
cvp/vid_encoder.py:132
↓ 5 callersMethod_apply_feat_constraint
(self, feat)
cvp/models.py:151
↓ 5 callersFunctionimagenet_deprocess_batch
Input: - imgs: FloatTensor of shape (N, C, H, W) giving preprocessed images Output: - imgs_de: ByteTensor of shape (N, C, H, W) givi
utils/data_utils.py:49
↓ 5 callersMethodparse
(self)
cfgs/base_cfgs.py:90
↓ 4 callersMethodadd_images
(self, ims, txts, links, width=400)
PerceptualSimilarity/util/html.py:35
↓ 4 callersFunctionbox_stack
(input)
utils/box_utils.py:226
↓ 4 callersFunctionbuild_pre_act
(input_dim, output_dim, activation='relu', batch_norm='batch', dropout=0)
cvp/layers.py:300
↓ 4 callersFunctionconvert_cv2torch
:param img_list: list of [numpy in shape of (H, W, C)] in BGR :return: list of FloatTensor [(C, H, W)] in RGB
cvp/vis.py:29
↓ 4 callersFunctionimagenet_preprocess
()
utils/data_utils.py:30
↓ 4 callersMethodinit_hidden
(self)
cvp/vid_encoder.py:299
↓ 4 callersFunctionmask_norm
:param logit: (O, 1, H, W) :return: softmax(O, 1, H, W)
cvp/layout.py:553
↓ 4 callersMethodsave_image
(self, image, save_dir, name)
cvp/evaluator.py:148
↓ 4 callersFunctionsplat
:param bbox: (O, 4) in [0, 1] :param feat: (O, D, h, w) :return: (O, D, H, W)
cvp/layout.py:415
↓ 4 callersFunctionzoom_to_res
(img,res=256,order=0,axis=0)
PerceptualSimilarity/util/visualizer.py:11
↓ 3 callersMethod__init__
(self, args)
cvp/models.py:221
↓ 3 callersMethod_forward_image_encode
(self, input)
cvp/models.py:155
↓ 3 callersMethod_forward_with_z
(self, vid_batch, img_z, time_len)
cvp/models.py:97
↓ 3 callersFunctionadd_layer_to
(last, output_dim, normalization, activation, rgb=3)
cvp/decoder.py:387
↓ 3 callersFunctionbuild_mlp
(dim_list, activation='relu', batch_norm='none', dropout=0, final_nonlinearity=False)
cvp/layers.py:215
↓ 3 callersFunctioncollate_batch
(batch_list, dt, V)
cvp/models.py:354
↓ 3 callersFunctionconvert_torch2cv
:param img: FloatTensor in shape (C, H, W) in RGB :return: uint8 numpy in shape (H, W, C) in BGR
cvp/vis.py:41
↓ 3 callersFunctionextents_to_centers
Convert boxes from [x0, y0, x1, y1] format to [xc, yc, w, h] format Input: - boxes: Input boxes of shape (N, 4) in [x0, y0, x1, y1] form
utils/box_utils.py:155
↓ 3 callersMethodfilter_time_stamp
(self, t, vid_batch)
cvp/models.py:158
↓ 3 callersMethodmu_logvar
(self, input_feat)
cvp/vid_encoder.py:76
↓ 3 callersFunctionvstack
(img_list)
cvp/evaluator.py:258
↓ 2 callersMethod__init__
(self, feat_dim_list, dims, radius, normalization='instance', activation='leakyrelu')
cvp/decoder.py:262
↓ 2 callersFunction_boxes_to_grid_inv
Input: - boxes: FloatTensor of shape (O, 4) giving boxes in the [x0, y0, x1, y1] in (0, 1) format in the [0, 1] coordinate space -
cvp/layout.py:591
↓ 2 callersMethod_forward
(self, vid_batch, sample=True)
cvp/vid_encoder.py:48
↓ 2 callersMethod_forward_tower
image: N, C, H, W
cvp/models.py:202
↓ 2 callersMethod_forward_tower
image: N, C, H, W
cvp/models.py:293
↓ 2 callersFunction_get_padding
Helper method to compute padding size
cvp/layers.py:119
↓ 2 callersFunction_init_conv
(layer, method)
cvp/layers.py:50
↓ 2 callersMethod_next_trip
(V, O, 2), (V, T, 3)
cvp/models.py:215
↓ 2 callersFunction_pool_samples
Input: - samples: FloatTensor of shape (O, D, H, W) - obj_to_img: LongTensor of shape (O,) with each element in the range [0, N) ma
cvp/layout.py:663
↓ 2 callersMethod_unsqz
(self, tensor, dim=0)
cvp/models.py:172
↓ 2 callersMethodadd_header
(self, str)
PerceptualSimilarity/util/html.py:27
↓ 2 callersFunctionadd_path
(path)
cvp/debug_init_paths.py:8
↓ 2 callersFunctionbilinear_sample
Perform bilinear sampling on the features in feats using the sampling grid given by X and Y. Inputs: - feats: Tensor holding input feature m
cvp/bilinear.py:190
↓ 2 callersFunctionbuild_vid_loaders
(args)
data/DemoImage.py:158
↓ 2 callersFunctioncrop_bbox
Take differentiable crops of feats specified by bbox. Inputs: - feats: Tensor of shape (N, C, H, W) - bbox: Bounding box coordinates of shap
cvp/bilinear.py:102
↓ 2 callersFunctioncrop_bbox_batch_cudnn
(feats, bbox, bbox_to_feats, HH, WW=None)
cvp/bilinear.py:68
↓ 2 callersMethodexpand_bg
(self, bg_feat, dt, V)
cvp/decoder.py:164
↓ 2 callersMethodexpand_bg
(self, bg_feat, dt, V)
cvp/decoder.py:382
↓ 2 callersMethodforward_inception
(self, intput, time_len, seed=None)
cvp/models.py:227
↓ 2 callersMethodforward_lp
todo: clean!
cvp/models.py:230
↓ 2 callersMethodget_bg_feat
(self, bg: list, t)
cvp/models.py:167
↓ 2 callersMethodget_image_dir
(self)
PerceptualSimilarity/util/html.py:24
↓ 2 callersFunctionim2tensor
(image, imtype=np.uint8, cent=1., factor=255./2.)
PerceptualSimilarity/util/util.py:142
↓ 2 callersMethodinitialize
(self)
cfgs/test_cfgs.py:12
↓ 2 callersFunctionmkdir
(path)
PerceptualSimilarity/util/util.py:250
↓ 2 callersFunctionnormalize_blob
(in_feat,eps=1e-10)
PerceptualSimilarity/util/util.py:61
↓ 2 callersFunctionnormalize_tensor
(in_feat,eps=1e-10)
PerceptualSimilarity/util/util.py:72
↓ 2 callersFunctionnp2tensor
(np_obj)
PerceptualSimilarity/util/util.py:93
↓ 2 callersMethodperceptual_metric
:param im0: (dt, V, 1, 3, H, W) in mean=(0, 0, 0) std=(1, 1, 1,)
cvp/evaluator.py:39
↓ 2 callersMethodreparameterize
generate sample from N(mu, var). or no sample
cvp/vid_encoder.py:65
↓ 2 callersFunctionresize_l1_loss
Resize gt to the size of pred. and calculate loss
cvp/losses.py:84
↓ 2 callersFunctionsave_image
(image_numpy, image_path, )
PerceptualSimilarity/util/util.py:196
↓ 2 callersMethodsave_raw_box_image
(self, obj, index, pref='', suff='')
cvp/evaluator.py:106
↓ 2 callersMethodsave_vid_traj
:param pred_list: list of image: [im0, im2, ...] :return:
cvp/evaluator.py:132
↓ 2 callersFunctionskeleton_10B
(box, img)
cvp/vis.py:217
↓ 2 callersFunctionskeleton_13B
(box, img)
cvp/vis.py:187
↓ 2 callersFunctionskeleton_13B_trip
:param box: (O, 2) :param img: :param trip: (T, 3) :return:
cvp/vis.py:132
↓ 2 callersFunctionskeleton_13B_valid
(box, img, valid)
cvp/vis.py:148
↓ 2 callersFunctionsplat_with_wgt
:param bbox: (O, 4) in [0, 1] :param feat: (O, D, h, w) :return: (O, D, H, W)
cvp/layout.py:318
↓ 2 callersFunctiontensor_linspace
Vectorized version of torch.linspace. Inputs: - start: Tensor of any shape - end: Tensor of the same shape as start - steps: Integer Re
cvp/bilinear.py:248
↓ 1 callersMethod__init__
(self, size)
cvp/layers.py:70
↓ 1 callersFunction_bbox_to_grid_fwd01
:param boxes: (O, 4) [x0, y0, x1, y1] in (0, 1) :return: (O, H, W, 2)
cvp/layout.py:571
↓ 1 callersMethod_build_graph
(self, index)
data/ShapeStacks.py:120
↓ 1 callersMethod_build_graph
(self, num_obj)
data/DemoImage.py:111
↓ 1 callersMethod_crop_image
(self, index, image, box_center)
data/ShapeStacks.py:106
↓ 1 callersMethod_crop_image
(self, index, image, box_center)
data/DemoImage.py:97
↓ 1 callersMethod_forward_bg
:param bg_feat: (N, Di, h, w) :param mask: (N, 1, H, W) :return: (N, Do, Hm, Wm)
cvp/decoder.py:147
↓ 1 callersMethod_forward_bg
.view(V, -1, bg_feat.size(-2), bg_feat.size(-1))
cvp/decoder.py:246
↓ 1 callersMethod_forward_bg
:param bg_feat: (N, Di, h, w) :param mask: (N, 1, H, W) :return: (N, Do, Hm, Wm)
cvp/decoder.py:365
↓ 1 callersMethod_forward_image_encode
(self, vid)
cvp/models.py:183
↓ 1 callersMethod_forward_n_step
(self, vid_batch, img_z, time_len)
cvp/models.py:133
↓ 1 callersMethod_gather_box
:param holder: (V, num, D) :param valid: (V, 1, O) :return: (V, O, D)
cvp/models.py:338
↓ 1 callersFunction_invperm
(p)
cvp/bilinear.py:61
↓ 1 callersMethod_next_trip
(self, bbox, cur_trip)
cvp/models.py:350
↓ 1 callersMethod_read_image
(self, index, bboxes)
data/ShapeStacks.py:98
↓ 1 callersMethod_read_image
(self, index, bboxes)
data/DemoImage.py:89
↓ 1 callersMethod_scatter_box
:param bbox: (V, O, D) :param valid: (V, 1, O) :return: (V, num, D)
cvp/models.py:326
↓ 1 callersMethodadd_histogram
(self, t, z, name='')
cvp/logger.py:40
↓ 1 callersMethodadd_loss
(self, t, dictionary, pref='train/')
cvp/logger.py:35
↓ 1 callersFunctionadd_path
(path)
_init_path.py:8
↓ 1 callersFunctionadd_path
(path)
data/debug_init_paths.py:8
↓ 1 callersMethodadd_table
(self, border=1)
PerceptualSimilarity/util/html.py:31
↓ 1 callersFunctionapply_box_transform
Apply box transforms to a set of anchor boxes. Inputs: - anchors: Anchor boxes of shape (N, 4), where each anchor is specified in
utils/box_utils.py:24
↓ 1 callersFunctionbatchNorm5d
(num_features, eps = 1e-5)
cvp/layers.py:343
next →1–100 of 349, ranked by callers