MCPcopy Create free account

hub / github.com/csjliang/PPR10K / functions

Functions88 in github.com/csjliang/PPR10K

↓ 10 callersFunctionread_img
read image by cv2 or from lmdb return: Numpy float32, HWC, BGR, [0,1]
code_3DLUT/datasets_GLC.py:185
↓ 8 callersMethod__init__
(self, dim=33)
code_3DLUT/models_x.py:268
↓ 7 callersFunctionchannel_convert
conversion among BGR, gray and y
code_3DLUT/datasets_GLC.py:200
↓ 3 callersFunction_is_numpy_image
(img)
code_3DLUT/torchvision_x_functional.py:47
↓ 3 callersFunctiongenerator
(img)
code_3DLUT/train_GLC.py:106
↓ 3 callersFunctionread_xmp_and_statis
(xmp_root)
utils/data_augment_get_target_xmp.py:42
↓ 2 callersMethodbackward
(ctx, lut_grad, x_grad)
code_3DLUT/models_x.py:231
↓ 2 callersFunctioncrop
crop image Arguments: img {ndarray} -- image to be croped top {int} -- top size left {int} -- left size hei
code_3DLUT/torchvision_x_functional.py:327
↓ 2 callersMethodforward
(self, LUT)
code_3DLUT/models_x.py:279
↓ 2 callersFunctionget_xmp_augment
(name_ori, name_aug, augment_param, xmp_file_root, pred_xmp_root)
utils/data_augment_get_target_xmp.py:4
↓ 1 callersFunctionTriLinearBackwardCpu
code_3DLUT/trilinear_cpp/src/trilinear.cpp:104
↓ 1 callersFunctionTriLinearForwardCpu
code_3DLUT/trilinear_cpp/src/trilinear.cpp:50
↓ 1 callersFunction_is_tensor_image
(img)
code_3DLUT/torchvision_x_functional.py:43
↓ 1 callersFunctioncalculate_psnr
()
code_3DLUT/train_GLC.py:125
↓ 1 callersFunctioncalculate_psnr
()
code_3DLUT/train.py:140
↓ 1 callersFunctiongenerator
(img)
code_3DLUT/validation.py:65
↓ 1 callersFunctiongenerator0
(img)
code_3DLUT/train.py:105
↓ 1 callersFunctiongenerator_batch
(img)
code_3DLUT/train.py:121
↓ 1 callersFunctiongetExif
(file_path, file_name)
utils/get_shot_time_and_plot_hist.py:10
↓ 1 callersFunctionget_GT_attribute_statistics
(num_steps, each_key)
utils/data_augment_get_target_xmp.py:78
↓ 1 callersFunctionget_augment_params_with_names
(num_files_aug_all, xmp_source_dict, source_xmp_root, aug_xmp_root, xmp_target_dict, target_xmp_root, aug_xmp_
utils/data_augment_get_target_xmp.py:108
↓ 1 callersFunctionget_augment_params_with_names
(num_files_aug_all, xmp_train_dict, xmp_file_root, aug_xmp_root)
utils/data_augment_get_xmps.py:109
↓ 1 callersFunctionget_xmp_augment
(name_ori, name_aug, augment_param, xmp_file_root, pred_xmp_root)
utils/data_augment_get_xmps.py:5
↓ 1 callersFunctionread_xmp_and_statis
(xmp_root)
utils/data_augment_get_xmps.py:43
↓ 1 callersFunctionresize
resize the image TODO: opencv resize 之后图像就成了0~1了 Arguments: img {ndarray} -- the input ndarray image size {int, iterable} -- t
code_3DLUT/torchvision_x_functional.py:269
↓ 1 callersFunctionvisualize_result
Saves a generated sample from the validation set
code_3DLUT/validation.py:84
FunctionPYBIND11_MODULE
code_3DLUT/trilinear_cpp/src/trilinear.cpp:170
FunctionPYBIND11_MODULE
code_3DLUT/trilinear_cpp/src/trilinear_cuda.cpp:31
Method__getitem__
(self, index)
code_3DLUT/datasets_GLC.py:44
Method__getitem__
(self, index)
code_3DLUT/datasets_evaluation.py:15
Method__getitem__
(self, index)
code_3DLUT/datasets.py:30
Method__init__
(self, root, mode="train", use_mask=False)
code_3DLUT/datasets_GLC.py:11
Method__init__
(self, root)
code_3DLUT/datasets_evaluation.py:10
Method__init__
(self, out_dim=5, aug_test=False)
code_3DLUT/models_x.py:19
Method__init__
(self, out_dim=5, aug_test=False)
code_3DLUT/models_x.py:42
Method__init__
(self, out_dim=5, aug_test=False)
code_3DLUT/models_x.py:67
Method__init__
(self, in_channels=3)
code_3DLUT/models_x.py:102
Method__init__
(self, out_dim=5)
code_3DLUT/models_x.py:124
Method__init__
(self, dim=33)
code_3DLUT/models_x.py:146
Method__init__
(self, dim=33)
code_3DLUT/models_x.py:173
Method__init__
(self)
code_3DLUT/models_x.py:261
Method__init__
(self, root, mode="train", use_mask=False)
code_3DLUT/datasets.py:16
Method__len__
(self)
code_3DLUT/datasets_GLC.py:179
Method__len__
(self)
code_3DLUT/datasets_evaluation.py:28
Method__len__
(self)
code_3DLUT/datasets.py:91
Functionadjust_brightness
(img, value=0)
code_3DLUT/torchvision_x_functional.py:185
Functionadjust_contrast
(img, factor)
code_3DLUT/torchvision_x_functional.py:199
Functionadjust_hue
()
code_3DLUT/torchvision_x_functional.py:216
Functionadjust_saturation
()
code_3DLUT/torchvision_x_functional.py:212
Functionbbox_crop
crop bbox Arguments: img {ndarray} -- image to be croped top {int} -- top size left {int} -- left size heig
code_3DLUT/torchvision_x_functional.py:518
Functionbbox_hflip
horizontal flip the bboxes ^ ............. . . . . . . . . . . . . .............
code_3DLUT/torchvision_x_functional.py:480
Functionbbox_pad
(bboxes, padding)
code_3DLUT/torchvision_x_functional.py:538
Functionbbox_resize
resize the bbox Args: bboxes (ndarray): bbox ndarray [box_nums, 4] img_size (tuple): the image height and width targe
code_3DLUT/torchvision_x_functional.py:500
Functionbbox_shift
(bboxes, top, left)
code_3DLUT/torchvision_x_functional.py:457
Functionbbox_vflip
vertical flip the bboxes ........... . . . . >...........< . . . . ........... Args:
code_3DLUT/torchvision_x_functional.py:461
Functioncenter_crop
crop image Arguments: img {ndarray} -- input image output_size {number or sequence} -- the output image size. if sequence, sh
code_3DLUT/torchvision_x_functional.py:360
Functionelastic_transform
Elastic deformation of images as described in [Simard2003]_ (with modifications). Based on https://gist.github.com/erniejunior/601cdf56d2b424757de
code_3DLUT/torchvision_x_functional.py:407
Functionflip
(img, flip_code)
code_3DLUT/torchvision_x_functional.py:403
Methodforward
(self, x)
code_3DLUT/models_x.py:31
Methodforward
(self, x)
code_3DLUT/models_x.py:55
Methodforward
(self, x, m)
code_3DLUT/models_x.py:87
Methodforward
(self, img_input)
code_3DLUT/models_x.py:119
Methodforward
(self, img_input)
code_3DLUT/models_x.py:142
Methodforward
(self, x)
code_3DLUT/models_x.py:166
Methodforward
(self, x)
code_3DLUT/models_x.py:180
Methodforward
(ctx, lut, x)
code_3DLUT/models_x.py:188
Methodforward
(self, lut, x)
code_3DLUT/models_x.py:264
Functiongaussian_blur
(img, kernel_size)
code_3DLUT/torchvision_x_functional.py:180
Functionget_GT_attribute_statistics
(num_steps, each_key)
utils/data_augment_get_xmps.py:79
Functionhflip
(img)
code_3DLUT/torchvision_x_functional.py:400
Functionnoise
TODO: Not good for uint16 data
code_3DLUT/torchvision_x_functional.py:117
Functionnormalize
Normalize a tensor image with mean and standard deviation. .. note:: This transform acts out of place by default, i.e., it does not mutat
code_3DLUT/torchvision_x_functional.py:90
Functionpad
(img, padding, fill=0, padding_mode='constant')
code_3DLUT/torchvision_x_functional.py:302
Functionpreserve_channel_dim
Preserve dummy channel dim.
code_3DLUT/torchvision_x_functional.py:30
Functionresized_crop
(img, top, left, height, width, size, interpolation=Image.BILINEAR)
code_3DLUT/torchvision_x_functional.py:391
Functionrotate
(img, angle, center=None, scale=1.0)
code_3DLUT/torchvision_x_functional.py:257
Functionshift
(img, top, left)
code_3DLUT/torchvision_x_functional.py:249
Functionto_grayscale
convert input ndarray image to gray sacle image. Arguments: img {ndarray} -- the input ndarray image Keyword Arguments:
code_3DLUT/torchvision_x_functional.py:222
Functionto_pil_image
(tensor)
code_3DLUT/torchvision_x_functional.py:80
Functionto_tensor
convert numpy.ndarray to torch tensor. \n if the image is uint8 , it will be divided by 255;\n if the image is uint16 , it will be div
code_3DLUT/torchvision_x_functional.py:51
Functionto_tiff_image
(tensor)
code_3DLUT/torchvision_x_functional.py:85
Functiontrilinear_backward
code_3DLUT/trilinear_cpp/src/trilinear.cpp:29
Functiontrilinear_backward_cuda
code_3DLUT/trilinear_cpp/src/trilinear_cuda.cpp:18
Functiontrilinear_forward
code_3DLUT/trilinear_cpp/src/trilinear.cpp:8
Functiontrilinear_forward_cuda
code_3DLUT/trilinear_cpp/src/trilinear_cuda.cpp:5
Functionvflip
(img)
code_3DLUT/torchvision_x_functional.py:397
Functionweights_init_normal_classifier
(m)
code_3DLUT/models_x.py:7
Functionwrapped_function
(img, *args, **kwargs)
code_3DLUT/torchvision_x_functional.py:33