Functions — github.com/MaureenZOU/TSAM

hub / github.com/MaureenZOU/TSAM / functions

Functions693 in github.com/MaureenZOU/TSAM

⨍ Functions 693◇ Types & classes 158

↓ 2 callersFunctionget_loader

(type, ratio)

src/scripts/dataset_json_generator.py:9

↓ 2 callersFunctionget_random_velocity

(max_speed=3, dist='uniform')

src/xy_scripts/stationary.py:120

↓ 2 callersFunctionget_random_velocity

(max_speed=3, dist='uniform')

src/utils/stationary_mask.py:120

↓ 2 callersMethodgram_matrix

(self, input)

src/model/loss.py:69

↓ 2 callersMethodgram_matrix

(self, input)

src/model/loss_module.py:80

↓ 2 callersFunctionim2tensor

(image, imtype=np.uint8, cent=1., factor=255./2.)

libs/PerceptualSimilarity/util/util.py:144

↓ 2 callersFunctionimg_read_yuv

(src_file, width, height)

libs/video_quality/measure.py:34

↓ 2 callersMethodinitialize

(self, datafolders, dataroot='./dataset',dataset_mode='2afc',load_size=64,batch_size=1,serial_batches=True, nT

libs/PerceptualSimilarity/data/custom_dataset_data_loader.py:23

↓ 2 callersFunctionmkdir

(path)

libs/PerceptualSimilarity/util/util.py:252

↓ 2 callersMethodname

(self)

libs/PerceptualSimilarity/models/dist_model.py:21

↓ 2 callersFunctionnormalize_blob

(in_feat,eps=1e-10)

libs/PerceptualSimilarity/util/util.py:63

↓ 2 callersFunctionnormalize_tensor

(in_feat,eps=1e-10)

libs/PerceptualSimilarity/util/util.py:74

↓ 2 callersFunctionnp2tensor

(np_obj)

libs/PerceptualSimilarity/util/util.py:95

↓ 2 callersMethodpreprocess

(self, masked_imgs, masks, guidances, flows)

src/model/networks.py:66

↓ 2 callersFunctionrandom_accelerate

(velocity, maxAcceleration, dist='uniform')

src/utils/mask_generators.py:91

↓ 2 callersFunctionread_filenames_from_dir

(dir_name, reader, max_length=None)

src/utils/util.py:247

↓ 2 callersFunctionresize_flow

(flow, size)

src/utils/visual_helper.py:116

↓ 2 callersFunctionresize_flow_np

(flow, size)

src/utils/visual_helper.py:127

↓ 2 callersFunctionrm_dirs

(dir_name, ignore_errors=False)

src/utils/util.py:53

↓ 2 callersMethodsave

(self, label)

libs/PerceptualSimilarity/models/base_model.py:31

↓ 2 callersFunctionsave_frames_to_dir

(frames, dirname)

src/utils/readers.py:570

↓ 2 callersMethodsave_network

(self, network, path, network_label, epoch_label)

libs/PerceptualSimilarity/models/base_model.py:35

↓ 2 callersMethodshift

(x, n_segment, fold_div=3, inplace=False)

src/model/tsm/temporal_shift.py:184

↓ 2 callersMethodsplit_validation

(self)

src/base/base_data_loader.py:52

↓ 2 callersMethodsummary

Model summary

src/base/base_model.py:22

↓ 2 callersFunctiontransform

(k, v)

src/scripts/FVI_checkpoint_compatibility_transform.py:58

↓ 2 callersFunctionwalk_clean

(root)

src/scripts/clean_saved.py:52

↓ 1 callersFunctionCreateDataset

(dataroots,dataset_mode='2afc',load_size=64,)

libs/PerceptualSimilarity/data/custom_dataset_data_loader.py:5

↓ 1 callersMethod_construct_diff_model

(self, base_model, keep_rgb=False)

src/model/tsm/models.py:326

↓ 1 callersMethod_construct_flow_model

(self, base_model)

src/model/tsm/models.py:291

↓ 1 callersMethod_eval_metrics

(self, output, target)

src/trainer/inference.py:296

↓ 1 callersMethod_evaluate_data_loader

(self, epoch=None, output_root_dir=None, data_loader=None, name='test')

src/trainer/inference.py:141

↓ 1 callersMethod_evaluate_data_loader

(self, epoch=None, output_root_dir=None, data_loader=None, name='test')

src/trainer/trainer.py:143

↓ 1 callersMethod_evaluate_test_video

(self, output, gt_frames, masks)

src/trainer/inference.py:285

↓ 1 callersMethod_evaluate_test_video

(self, output, gt_frames, masks)

src/trainer/trainer.py:293

↓ 1 callersMethod_forward_impl

(self, x, flows)

src/model/tsm/resnet.py:211

↓ 1 callersMethod_get_loss

(self, outputs, warped_outputs, non_occlusion_masks, masks)

src/model/loss.py:235

↓ 1 callersMethod_get_loss

(self, outputs, warped_outputs, non_occlusion_masks, masks)

src/model/loss_module.py:243

↓ 1 callersMethod_get_mask_name

(self, index)

src/data_loader/dataset.py:328

↓ 1 callersMethod_get_mask_name

(self, index)

src/data_loader/dataset.py:521

↓ 1 callersMethod_get_mask_name

(self, index)

src/data_loader/dataset.py:709

↓ 1 callersMethod_get_mask_name

(self, index)

src/data_loader/dataset.py:902

↓ 1 callersMethod_get_masks

(self, size, start_idx, end_idx, fg_dir)

src/data_loader/dataset.py:90

↓ 1 callersMethod_get_masks

(self, size, start_idx, end_idx, fg_dir)

src/data_loader/dataset.py:244

↓ 1 callersMethod_get_masks

(self, size, start_idx, end_idx, fg_dir)

src/data_loader/dataset.py:410

↓ 1 callersMethod_get_masks

(self, size, start_idx, end_idx, fg_dir)

src/data_loader/dataset.py:602

↓ 1 callersMethod_get_masks

(self, size, start_idx, end_idx, fg_dir)

src/data_loader/dataset.py:791

↓ 1 callersMethod_get_non_gan_loss

(self, non_gan_loss)

src/trainer/inference.py:403

↓ 1 callersMethod_get_non_occlusion_mask

(self, targets, warped_targets)

src/model/loss.py:227

↓ 1 callersMethod_get_non_occlusion_mask

(self, targets, warped_targets)

src/model/loss_module.py:235

↓ 1 callersMethod_get_sample_index_from_video

(self, length)

src/data_loader/dataset.py:81

↓ 1 callersMethod_get_sample_index_from_video

(self, length)

src/data_loader/dataset.py:235

↓ 1 callersMethod_get_sample_index_from_video

(self, length)

src/data_loader/dataset.py:401

↓ 1 callersMethod_get_sample_index_from_video

(self, length)

src/data_loader/dataset.py:593

↓ 1 callersMethod_get_sample_index_from_video

(self, length)

src/data_loader/dataset.py:782

↓ 1 callersMethod_get_spatial_sr_mask

(self)

src/data_loader/dataset.py:1010

↓ 1 callersMethod_load_pretrained

(self, pretrained_path)

src/base/base_trainer.py:207

↓ 1 callersMethod_load_pretrained

(self, pretrained_path)

src/base/base_inference.py:208

↓ 1 callersMethod_prepare_base_model

(self, base_model)

src/model/tsm/models.py:107

↓ 1 callersMethod_prepare_device

setup GPU device if available, move model into configured device

src/base/base_trainer.py:75

↓ 1 callersMethod_prepare_device

setup GPU device if available, move model into configured device

src/base/base_inference.py:75

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:119

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:273

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:439

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:631

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:821

↓ 1 callersMethod_resume_checkpoint

Resume from saved checkpoints :param resume_path: Checkpoint path to be resumed

src/base/base_trainer.py:179

↓ 1 callersMethod_resume_checkpoint

Resume from saved checkpoints :param resume_path: Checkpoint path to be resumed

src/base/base_inference.py:180

↓ 1 callersMethod_sample_crop_size

(self, im_size)

src/data_loader/transform_flow.py:217

↓ 1 callersMethod_sample_crop_size

(self, im_size)

src/data_loader/transform.py:155

↓ 1 callersMethod_sample_fix_offset

(self, image_w, image_h, crop_w, crop_h)

src/data_loader/transform_flow.py:241

↓ 1 callersMethod_sample_fix_offset

(self, image_w, image_h, crop_w, crop_h)

src/data_loader/transform.py:179

↓ 1 callersMethod_save_checkpoint

Saving checkpoints :param epoch: current epoch number :param log: logging information of the epoch :param save_best:

src/base/base_trainer.py:145

↓ 1 callersMethod_save_checkpoint

Saving checkpoints :param epoch: current epoch number :param log: logging information of the epoch :param save_best:

src/base/base_inference.py:146

↓ 1 callersMethod_save_file

(self, output_dir, i, file_)

src/utils/readers.py:140

↓ 1 callersMethod_setup

(self)

src/model/loss.py:223

↓ 1 callersMethod_setup

(self)

src/model/loss_module.py:231

↓ 1 callersMethod_split_sampler

(self, split)

src/base/base_data_loader.py:29

↓ 1 callersMethod_store_gated_values

(self, out_dir)

src/trainer/inference.py:60

↓ 1 callersMethod_store_gated_values

(self, out_dir)

src/trainer/trainer.py:60

↓ 1 callersMethod_train_epoch

Training logic for an epoch :param epoch: Current epoch number

src/base/base_trainer.py:137

↓ 1 callersMethod_train_epoch

Training logic for an epoch :param epoch: Current epoch number

src/base/base_inference.py:138

↓ 1 callersMethod_valid_epoch

Validate after training an epoch :return: A log that contains information about validation Note: The validation

src/trainer/trainer.py:557

↓ 1 callersFunctionadd_course_net_structures

(checkpoint)

src/scripts/FVI_checkpoint_compatibility_transform.py:47

↓ 1 callersMethodadd_table

(self, border=1)

libs/PerceptualSimilarity/util/html.py:31

↓ 1 callersFunctionapply_video

(video_dir, mask_dir, out_dir)

src/scripts/mask_scripts/apply_mask.py:34

↓ 1 callersFunctionask_all_in_once

(args, collected)

src/scripts/clean_saved.py:63

↓ 1 callersFunctionask_one_by_one

(args, collected)

src/scripts/clean_saved.py:31

↓ 1 callersMethodbackward_train

(self)

libs/PerceptualSimilarity/models/dist_model.py:197

↓ 1 callersFunctionbbox_offset

(bbox, location)

src/utils/util.py:70

↓ 1 callersMethodbuild

(self)

src/model/i3d.py:311

↓ 1 callersFunctioncalculate_frechet_distance

Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2

src/evaluate.py:391

↓ 1 callersMethodclamp_weights

(self)

libs/PerceptualSimilarity/models/dist_model.py:165

↓ 1 callersFunctionclean_empty_exp

(args)

src/scripts/clean_saved.py:51

↓ 1 callersFunctioncollect_satisfied

(args)

src/scripts/clean_saved.py:14

↓ 1 callersMethodcombine_videos

(self, videos, col=2, edge_offset=35, h_start_offset=35)

src/utils/readers.py:513

↓ 1 callersFunctioncompress_flow

(flow, bound=96)

src/utils/visual_helper.py:174

↓ 1 callersFunctioncomputeColor

(u, v)

src/utils/visual_helper.py:80

↓ 1 callersMethodcompute_accuracy

d0, d1 are Variables, judge is a Tensor

libs/PerceptualSimilarity/models/dist_model.py:200

↓ 1 callersFunctioncompute_color

compute optical flow color map :param u: optical flow horizontal map :param v: optical flow vertical map :return: optical flow in col

src/utils/flow_utils.py:85

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