MCPcopy Create free account

hub / github.com/YZY-stack/DF40 / functions

Functions1,098 in github.com/YZY-stack/DF40

↓ 39 callersMethodload
(self, epoch, model_dir)
DeepfakeBench_DF40/training/networks/xception_ffd.py:259
↓ 24 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
DeepfakeBench_DF40/training/networks/adaface.py:214
↓ 23 callersMethodsave
(self, epoch, optim, model_dir)
DeepfakeBench_DF40/training/networks/xception_ffd.py:254
↓ 17 callersMethodstep
(self, closure=None)
DeepfakeBench_DF40/training/optimizor/SAM.py:59
↓ 16 callersMethodupdate
(self, item, num=1)
DeepfakeBench_DF40/training/metrics/base_metrics_class.py:195
↓ 14 callersMethodreset
Reset the Meter.
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/meters.py:141
↓ 13 callersMethodparse
(img, reg, landmarks)
DeepfakeBench_DF40/training/dataset/utils/SLADD.py:136
↓ 12 callersMethod__init__
The `__init__` method of any subclass should also contain these arguments. Args: cfg (CfgNode): model buildi
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:404
↓ 12 callersFunctionget_norm
Args: cfg (CfgNode): model building configs, details are in the comments of the config file. Returns: nn.Module:
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/batchnorm_helper.py:15
↓ 12 callersMethodinit_weights
This method can be optionally implemented by subclasses.
DeepfakeBench_DF40/training/networks/base_backbone.py:28
↓ 12 callersMethodnormalize
Normalize an image.
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:390
↓ 11 callersMethod__init__
(self, learnable=False)
DeepfakeBench_DF40/training/detectors/srm_detector.py:183
↓ 11 callersMethodtrain
(self, mode=True)
DeepfakeBench_DF40/training/networks/vgg.py:125
↓ 10 callersMethodclear
(self)
DeepfakeBench_DF40/training/metrics/base_metrics_class.py:203
↓ 9 callersMethodbackward
Perform backwarding, gathering the gradients across different process/ GPU group.
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/batchnorm_helper.py:146
↓ 9 callersFunctionconv3x3
3x3 convolution with padding
DeepfakeBench_DF40/training/networks/resnet.py:28
↓ 8 callersMethod__init__
(self, *, image_size, patch_size, num_classes, dim, depth, heads, mlp_dim, pool = 'cls', channels = 3, dim_hea
DeepfakeBench_DF40/training/networks/time_transformer.py:92
↓ 8 callersMethod__init__
(self, in_c, embedding_size)
DeepfakeBench_DF40/training/networks/adaface.py:94
↓ 8 callersMethod__init__
(self, eps=1e-5)
DeepfakeBench_DF40/training/networks/resnet.py:35
↓ 8 callersMethodadd_value
Add a new scalar value to the deque.
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/meters.py:399
↓ 8 callersFunctioncalculate_metrics_for_train
(label, output)
DeepfakeBench_DF40/training/metrics/base_metrics_class.py:24
↓ 8 callersFunctionconv1x1
1x1 convolution
DeepfakeBench_DF40/training/networks/iresnet.py:23
↓ 8 callersMethodencode
(self, x)
DeepfakeBench_DF40/training/dataset/utils/image_ae.py:45
↓ 8 callersMethodget_win_median
Calculate the current median value of the deque.
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/meters.py:407
↓ 8 callersMethodto_tensor
Convert an image to a PyTorch tensor.
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:384
↓ 7 callersFunctionget_logger
Retrieve the logger with the specified name or, if name is None, return a logger which is the root logger of the hierarchy. Args:
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/logging.py:71
↓ 7 callersFunctionget_point
(idx)
DeepfakeBench_DF40/training/dataset/utils/face_blend.py:46
↓ 5 callersMethod__init__
(self, c_in, c_out, reps, stride=1, start_with_relu=True, grow_first=True)
DeepfakeBench_DF40/training/networks/xception_ffd.py:20
↓ 5 callersMethod__init__
(self, in_dim, out_dim=None, add=False, ratio=8)
DeepfakeBench_DF40/training/lib/component/attention.py:54
↓ 5 callersFunction_iresnet
(arch, block, layers, pretrained, progress, **kwargs)
DeepfakeBench_DF40/training/networks/iresnet.py:162
↓ 5 callersFunction_iresnet
(arch, block, layers, pretrained, progress, **kwargs)
DeepfakeBench_DF40/training/networks/iresnet_iid.py:166
↓ 5 callersMethodapply
(self, img, interpolation_down=cv2.INTER_AREA, interpolation_up=cv2.INTER_CUBIC, **params)
DeepfakeBench_DF40/training/dataset/albu.py:34
↓ 5 callersFunctionget_checkpoint_dir
Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/checkpoint.py:37
↓ 5 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2232
↓ 5 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2466
↓ 5 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2700
↓ 5 callersMethodget_losses
(self, data_dict: dict, pred_dict: dict)
DeepfakeBench_DF40/training/detectors/sbi_detector.py:89
↓ 5 callersFunctionload_checkpoint
Load the checkpoint from the given file. If inflation is True, inflate the 2D Conv weights from the checkpoint to 3D Conv. Args:
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/checkpoint.py:223
↓ 5 callersMethodload_landmark
Load 2D facial landmarks from a file path. Args: file_path: A string indicating the path to the landmark file.
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:355
↓ 5 callersMethodload_rgb
Load an RGB image from a file path and resize it to a specified resolution. Args: file_path: A string indicating the pat
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:282
↓ 5 callersMethodupsample
(self, x, dims=["space"])
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1298
↓ 4 callersMethod__init__
(self)
DeepfakeBench_DF40/training/dataset/utils/image_ae.py:101
↓ 4 callersMethod__init__
ResBlock class constructs redisual blocks. More details can be found in: Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun.
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/resnet_helper.py:334
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
DeepfakeBench_DF40/training/networks/iresnet.py:121
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
DeepfakeBench_DF40/training/networks/iresnet_iid.py:123
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
DeepfakeBench_DF40/training/networks/resnet.py:295
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
DeepfakeBench_DF40/training/networks/resnet.py:379
↓ 4 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1309
↓ 4 callersMethodcontrastive_loss
(self, anchor, positive, negative)
DeepfakeBench_DF40/training/loss/contrastive_regularization.py:44
↓ 4 callersFunctioncreate_bbox_face
(image, landmarks, margin=0)
DeepfakeBench_DF40/training/dataset/sbi_dataset.py:154
↓ 4 callersFunctiondist
(p1, p2)
DeepfakeBench_DF40/training/dataset/utils/attribution_mask.py:7
↓ 4 callersFunctionget_5_keypoint
(shape)
DeepfakeBench_DF40/training/dataset/utils/face_blend.py:45
↓ 4 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1533
↓ 4 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1765
↓ 4 callersMethodget_detach_var
(self,x)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1999
↓ 4 callersFunctionget_test_metrics
(y_pred, y_true, img_names)
DeepfakeBench_DF40/training/metrics/utils.py:30
↓ 4 callersMethodget_writer
(self, phase, dataset_key, metric_key)
DeepfakeBench_DF40/training/trainer/trainer.py:85
↓ 4 callersFunctionremove_eyes
(image, landmarks, opt)
DeepfakeBench_DF40/training/dataset/utils/SLADD.py:46
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
DeepfakeBench_DF40/training/networks/cls_hrnet.py:46
↓ 3 callersMethod__init__
(self, func)
DeepfakeBench_DF40/training/networks/vgg.py:12
↓ 3 callersMethod__init__
(self, in_filters, out_filters, reps, strides=1, start_with_relu=True, grow_first=True)
DeepfakeBench_DF40/training/networks/xception_sladd.py:58
↓ 3 callersMethod__init__
(self, num_classes, drop_rate=0.2)
DeepfakeBench_DF40/training/detectors/recce_detector.py:121
↓ 3 callersMethod__init__
( self, in_channels, out_channels, kernel_size, padding=0, str
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/unet_helper.py:7
↓ 3 callersMethod__init__
(self, func)
DeepfakeBench_DF40/training/loss/vgg_loss.py:19
↓ 3 callersMethod__init__
(self)
DeepfakeBench_DF40/training/loss/classNseg_loss.py:9
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
DeepfakeBench_DF40/training/networks/cls_hrnet.py:423
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
DeepfakeBench_DF40/training/networks/cls_hrnet.py:370
↓ 3 callersFunction_scale_array
(arr, clip=True)
DeepfakeBench_DF40/training/dataset/utils/color_transfer.py:326
↓ 3 callersFunction_scale_array
(arr, clip=True)
DeepfakeBench_DF40/training/detectors/utils/color_transfer.py:326
↓ 3 callersMethodaverage
(self)
DeepfakeBench_DF40/training/metrics/base_metrics_class.py:199
↓ 3 callersFunctionchannel_hist_match
(source, template, hist_match_threshold=255, mask=None)
DeepfakeBench_DF40/training/dataset/utils/color_transfer.py:340
↓ 3 callersFunctionchannel_hist_match
(source, template, hist_match_threshold=255, mask=None)
DeepfakeBench_DF40/training/detectors/utils/color_transfer.py:340
↓ 3 callersFunctioncolor_transfer_sot
Color Transform via Sliced Optimal Transfer ported by @iperov from https://github.com/dcoeurjo/OTColorTransfer src - any float r
DeepfakeBench_DF40/training/dataset/utils/color_transfer.py:10
↓ 3 callersFunctioncolor_transfer_sot
Color Transform via Sliced Optimal Transfer ported by @iperov from https://github.com/dcoeurjo/OTColorTransfer src - any flo
DeepfakeBench_DF40/training/detectors/utils/color_transfer.py:10
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1530
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1762
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1996
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2229
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2463
↓ 3 callersMethodconcat
(self,x,y)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2697
↓ 3 callersMethoddata_aug
Apply data augmentation to an image, landmark, and mask. Args: img: An Image object containing the image to be augmented
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:399
↓ 3 callersFunctiondist
(a, b)
DeepfakeBench_DF40/training/dataset/utils/SLADD.py:11
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1292
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1513
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1745
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:1979
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2213
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2447
↓ 3 callersMethoddual_define
(self,name,labels,net)
DeepfakeBench_DF40/training/detectors/utils/slowfast/models/video_model_builder.py:2681
↓ 3 callersMethodfea_part2
(self, x)
DeepfakeBench_DF40/training/networks/xception.py:212
↓ 3 callersMethodfea_part3
(self, x)
DeepfakeBench_DF40/training/networks/xception.py:219
↓ 3 callersMethodfea_part4
(self, x)
DeepfakeBench_DF40/training/networks/xception.py:229
↓ 3 callersMethodfea_part5
(self, x)
DeepfakeBench_DF40/training/networks/xception.py:240
↓ 3 callersFunctionget_blend_mask
(mask)
DeepfakeBench_DF40/training/dataset/sbi_api.py:43
↓ 3 callersFunctionget_hull_mask
(img, shape, mtype='hull')
DeepfakeBench_DF40/training/dataset/utils/face_blend.py:143
↓ 3 callersMethodload_mask
Load a binary mask image from a file path and resize it to a specified resolution. Args: file_path: A string indicating
DeepfakeBench_DF40/training/dataset/abstract_dataset.py:319
↓ 3 callersMethodnorm_n_corr
(self, x)
DeepfakeBench_DF40/training/detectors/recce_detector.py:158
↓ 3 callersFunctionrandom_get_hull
(landmark,img1,hull_type=None)
DeepfakeBench_DF40/training/dataset/utils/bi_online_generation.py:57
↓ 3 callersFunctionread_csv
Loads boxes and class labels from a CSV file in the AVA format. CSV file format described at https://research.google.com/ava/download.html. Ar
DeepfakeBench_DF40/training/detectors/utils/slowfast/utils/ava_eval_helper.py:53
↓ 3 callersFunctionremove_mouth
(image, landmarks)
DeepfakeBench_DF40/training/dataset/utils/SLADD.py:79
next →1–100 of 1,098, ranked by callers