Functions — github.com/amjltc295/Free-Form-Video-Inpainting

hub / github.com/amjltc295/Free-Form-Video-Inpainting / functions

Functions471 in github.com/amjltc295/Free-Form-Video-Inpainting

⨍ Functions 471◇ Types & classes 106

↓ 1 callersMethod_get_spatial_sr_mask

(self)

src/data_loader/dataset.py:264

↓ 1 callersMethod_load_pretrained

(self, pretrained_path)

src/base/base_trainer.py:206

↓ 1 callersMethod_prepare_device

setup GPU device if available, move model into configured device

src/base/base_trainer.py:75

↓ 1 callersMethod_process_vds

(self, vds)

src/data_loader/dataset.py:106

↓ 1 callersMethod_resume_checkpoint

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

src/base/base_trainer.py:178

↓ 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.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:144

↓ 1 callersMethod_save_file

(self, output_dir, i, file_)

src/utils/readers.py:137

↓ 1 callersMethod_setup

(self)

src/model/loss.py:223

↓ 1 callersMethod_split_sampler

(self, split)

src/base/base_data_loader.py:29

↓ 1 callersMethod_store_gated_values

(self, out_dir)

src/trainer/trainer.py:55

↓ 1 callersMethod_train_epoch

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

src/base/base_trainer.py:136

↓ 1 callersMethod_valid_epoch

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

src/trainer/trainer.py:442

↓ 1 callersFunctionadd_course_net_structures

(checkpoint)

src/scripts/FVI_checkpoint_compatibility_transform.py:47

↓ 1 callersMethodadd_entry

(self, entry)

src/utils/logger.py:13

↓ 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:353

↓ 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:445

↓ 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

↓ 1 callersFunctioncompute_features

(img_norm)

libs/video_quality/niqe.py:104

↓ 1 callersFunctionconnect_landmarks

(image_shape, landmarks, color=None, alpha=0.75)

src/utils/face.py:60

↓ 1 callersFunctioncopy_masks_without_boarder

(root_dir, args)

src/scripts/mask_scripts/gen_masks.py:152

↓ 1 callersMethoddisplay_current_results

(self, visuals, epoch, nrows=None, res=256)

libs/PerceptualSimilarity/util/visualizer.py:43

↓ 1 callersMethodedge_loss

(self, output, target)

src/model/loss.py:122

↓ 1 callersFunctionevaluate_all_videos

( root_gt_dir, root_result_dir, flownet_checkpoint_path, root_mask_dir, test_num, result_postfix="

src/evaluate.py:205

↓ 1 callersFunctionevaluate_image

(gt, result)

src/evaluate.py:109

↓ 1 callersFunctionevaluate_video

( result_video_dir: str, gt_video_dir: str, video_mask_dir: str, flownet_checkpoint_path: str

src/evaluate.py:124

↓ 1 callersMethodextract_features

(self, x, target_endpoint='Logits')

src/model/i3d.py:326

↓ 1 callersFunctionflow_to_image

Convert flow into middlebury color code image :param flow: optical flow map :return: optical flow image in middlebury color

src/utils/flow_utils.py:47

↓ 1 callersMethodforward

(self, x)

src/model/i3d.py:75

↓ 1 callersMethodforward_train

(self)

libs/PerceptualSimilarity/models/dist_model.py:186

↓ 1 callersMethodgated

(self, mask)

src/model/blocks.py:275

↓ 1 callersFunctiongen_masks

(in_direc, out_direc, args)

src/scripts/mask_scripts/gen_foresics_masks.py:56

↓ 1 callersFunctiongeneralized_gaussian_ratio_inverse

(k)

libs/video_quality/niqe.py:62

↓ 1 callersMethodgenerate_fg_frames

(self, frames, segms)

src/utils/readers.py:415

↓ 1 callersMethodgenerate_mask

(self, i)

src/utils/readers.py:382

↓ 1 callersMethodgenerate_masks

(self)

src/utils/readers.py:375

↓ 1 callersMethodget_current_errors

(self)

libs/PerceptualSimilarity/models/dist_model.py:206

↓ 1 callersMethodget_current_visuals

(self)

libs/PerceptualSimilarity/models/dist_model.py:215

↓ 1 callersMethodget_mask

(self, i)

src/utils/readers.py:412

↓ 1 callersFunctionget_random_mask

(img_size=[128, 128], r=[0.375, 0.5])

src/scripts/mask_scripts/gen_foresics_masks.py:39

↓ 1 callersFunctionget_random_stroke_control_points

Implementation the free-form training masks generating algorithm proposed by JIAHUI YU et al. in "Free-Form Image Inpainting with Gated Convo

src/utils/mask_generators.py:127

↓ 1 callersFunctionget_random_velocity

(max_speed, dist='uniform')

src/utils/mask_generators.py:166

↓ 1 callersFunctionget_stroke_preset

(stroke_preset)

src/scripts/mask_scripts/gen_masks.py:87

↓ 1 callersFunctionget_test_data_loaders

(types, ratios)

src/scripts/dataset_json_generator.py:35

↓ 1 callersFunctioninit_dm_model

()

src/evaluate.py:51

↓ 1 callersFunctioninit_warping_model

(checkpoint_path)

src/evaluate.py:28

↓ 1 callersFunctionis_image_file

(filename, mode='img')

libs/PerceptualSimilarity/data/image_folder.py:21

↓ 1 callersFunctionlandmarks_to_np

(landmarks, dtype="int")

src/utils/face.py:47

↓ 1 callersFunctionmain

(config, resume, output_root_dir=None, pretrained_path=None)

src/train.py:23

↓ 1 callersFunctionmain

(args)

src/evaluate.py:283

↓ 1 callersFunctionmain

(args)

src/scripts/compare_results.py:78

↓ 1 callersFunctionmain

()

src/scripts/FVI_checkpoint_compatibility_transform.py:80

↓ 1 callersFunctionmain

(args)

src/scripts/gen_output_videos.py:38

↓ 1 callersFunctionmain

(args)

src/scripts/check_len.py:31

↓ 1 callersFunctionmain

(args)

src/scripts/extract_frames.py:70

↓ 1 callersFunctionmain

(args)

src/scripts/rename.py:29

↓ 1 callersFunctionmain

(args)

src/scripts/mask_scripts/gen_masks.py:226

↓ 1 callersFunctionmake_color_wheel

Generate color wheel according Middlebury color code :return: Color wheel

src/utils/flow_utils.py:129

↓ 1 callersFunctionmake_dataset

(dirs, mode='img')

libs/PerceptualSimilarity/data/image_folder.py:27

↓ 1 callersFunctionmean_squared_error

(A, B)

src/utils/util.py:93

↓ 1 callersFunctionnormalize_image

(img, sigma=7/6)

libs/video_quality/niqe.py:118

↓ 1 callersFunctionnum_checkpoints

(path)

src/scripts/clean_saved.py:9

↓ 1 callersMethodoptimize_parameters

(self)

libs/PerceptualSimilarity/models/dist_model.py:158

↓ 1 callersFunctionoverride_data_setting

(config, dataset_config)

src/train.py:115

↓ 1 callersFunctionparse_args

()

src/evaluate.py:61

↓ 1 callersFunctionparse_args

()

src/utils/readers.py:20

↓ 1 callersFunctionparse_args

()

src/scripts/reorder.py:16

↓ 1 callersFunctionparse_args

()

src/scripts/leave_32frames.py:10

↓ 1 callersFunctionparse_args

()

src/scripts/compare_results.py:13

↓ 1 callersFunctionparse_args

()

src/scripts/clean_files.py:11

↓ 1 callersFunctionparse_args

()

src/scripts/FVI_checkpoint_compatibility_transform.py:8

↓ 1 callersFunctionparse_args

()

src/scripts/gen_output_videos.py:11

↓ 1 callersFunctionparse_args

()

src/scripts/check_len.py:9

↓ 1 callersFunctionparse_args

()

src/scripts/extract_frames.py:14

↓ 1 callersFunctionparse_args

()

src/scripts/rename.py:5

↓ 1 callersFunctionparse_args

()

src/scripts/mask_scripts/merge_into_base.py:16

↓ 1 callersFunctionparse_args

()

src/scripts/mask_scripts/copy_from_base.py:11

↓ 1 callersFunctionparse_args

()

src/scripts/mask_scripts/apply_mask.py:12

↓ 1 callersFunctionparse_args

()

src/scripts/mask_scripts/gen_foresics_masks.py:14

↓ 1 callersFunctionparse_args

()

src/scripts/mask_scripts/gen_masks.py:16

↓ 1 callersMethodplot_current_errors

(self, epoch, counter_ratio, opt, errors)

libs/PerceptualSimilarity/util/visualizer.py:126

↓ 1 callersMethodplot_current_errors_save

(self, epoch, counter_ratio, opt, errors,keys='+ALL',name='loss', to_plot=False)

libs/PerceptualSimilarity/util/visualizer.py:94

↓ 1 callersMethodprint_current_errors

(self, epoch, i, errors, t, t2=-1, t2o=-1, fid=None)

libs/PerceptualSimilarity/util/visualizer.py:142

↓ 1 callersFunctionrandom_move_control_points

(Xs, Ys, lineVelocity, nMovePointRatio, maxPiontMove, maxLineAcceleration, boarderGap=15)

src/utils/mask_generators.py:107

↓ 1 callersMethodread_file

(self, filename)

src/utils/readers.py:140

↓ 1 callersFunctionread_filepaths

(args)

src/scripts/extract_frames.py:52

↓ 1 callersMethodread_files

(self, filenames)

src/utils/readers.py:122

↓ 1 callersFunctionrename_conv_to_featureConv

(checkpoint)

src/scripts/FVI_checkpoint_compatibility_transform.py:30

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