Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/arnoweng/PyTorch-Deep-Image-Steganography
/ functions
Functions
82 in github.com/arnoweng/PyTorch-Deep-Image-Steganography
⨍
Functions
82
◇
Types & classes
23
↓ 12 callers
Function
_is_pil_image
(img)
utils/transformed.py:26
↓ 9 callers
Function
print_log
(log_info, log_path, console=True)
main.py:477
↓ 9 callers
Method
update
(self, val, n=1)
main.py:522
↓ 3 callers
Function
crop
Crop the given PIL Image. Args: img (PIL Image): Image to be cropped. i: Upper pixel coordinate. j: Left pixel coordinate.
utils/transformed.py:237
↓ 3 callers
Function
five_crop
Crop the given PIL Image into four corners and the central crop. .. Note:: This transform returns a tuple of images and there may be a
utils/transformed.py:301
↓ 3 callers
Function
resize
Resize the input PIL Image to the given size. Args: img (PIL Image): Image to be resized. size (sequence or int): Desired output s
utils/transformed.py:167
↓ 3 callers
Function
save_result_pic
(this_batch_size, originalLabelv, ContainerImg, secretLabelv, RevSecImg, epoch, i, save_path)
main.py:493
↓ 2 callers
Method
__init__
(self, padding, fill=0)
utils/transformed.py:666
↓ 2 callers
Function
_is_numpy_image
(img)
utils/transformed.py:37
↓ 2 callers
Function
_is_tensor_image
(img)
utils/transformed.py:33
↓ 2 callers
Function
hflip
Horizontally flip the given PIL Image. Args: img (PIL Image): Image to be flipped. Returns: PIL Image: Horizontall flipped im
utils/transformed.py:275
↓ 2 callers
Function
pad
Pad the given PIL Image on all sides with the given "pad" value. Args: img (PIL Image): Image to be padded. padding (int or tuple)
utils/transformed.py:208
↓ 2 callers
Function
pil_loader
(path)
data/ImageFolderDataset.py:48
↓ 2 callers
Function
print_network
(net)
main.py:96
↓ 2 callers
Function
vflip
Vertically flip the given PIL Image. Args: img (PIL Image): Image to be flipped. Returns: PIL Image: Vertically flipped image
utils/transformed.py:288
↓ 1 callers
Method
__init__
(self, input_nc, output_nc, num_downs, ngf=64, norm_layer=nn.BatchNorm2d, use_dropout=False,
models/HidingUNet.py:20
↓ 1 callers
Function
accimage_loader
(path)
data/ImageFolderDataset.py:55
↓ 1 callers
Function
adjust_brightness
Adjust brightness of an Image. Args: img (PIL Image): PIL Image to be adjusted. brightness_factor (float): How much to adjust the
utils/transformed.py:365
↓ 1 callers
Function
adjust_contrast
Adjust contrast of an Image. Args: img (PIL Image): PIL Image to be adjusted. contrast_factor (float): How much to adjust the cont
utils/transformed.py:383
↓ 1 callers
Function
adjust_hue
Adjust hue of an image. The image hue is adjusted by converting the image to HSV and cyclically shifting the intensities in the hue channel (H
utils/transformed.py:419
↓ 1 callers
Function
adjust_saturation
Adjust color saturation of an image. Args: img (PIL Image): PIL Image to be adjusted. saturation_factor (float): How much to adju
utils/transformed.py:401
↓ 1 callers
Method
get_params
Get parameters for ``crop`` for center crop. Args: img (PIL Image): Image to be cropped. output_size (tuple): Expected
utils/transformed.py:629
↓ 1 callers
Method
get_params
Get parameters for ``crop`` for a random crop. Args: img (PIL Image): Image to be cropped. output_size (tuple): Expect
utils/transformed.py:720
↓ 1 callers
Method
get_params
Get parameters for ``crop`` for a random sized crop. Args: img (PIL Image): Image to be cropped. Returns: tupl
utils/transformed.py:798
↓ 1 callers
Method
get_params
Get a randomized transform to be applied on image. Arguments are same as that of __init__. Returns: Transform which random
utils/transformed.py:954
↓ 1 callers
Function
is_image_file
Checks if a file is an image. Args: filename (string): path to a file Returns: bool: True if the filename ends with a known
data/ImageFolderDataset.py:10
↓ 1 callers
Function
main
()
main.py:125
↓ 1 callers
Function
normalize
Normalize a tensor image with mean and standard deviation. See ``Normalize`` for more details. Args: tensor (Tensor): Tensor image of
utils/transformed.py:149
↓ 1 callers
Method
reset
(self)
main.py:516
↓ 1 callers
Function
resized_crop
Crop the given PIL Image and resize it to desired size. Notably used in RandomResizedCrop. Args: img (PIL Image): Image to be cropped.
utils/transformed.py:254
↓ 1 callers
Function
save_current_codes
(des_path)
main.py:105
↓ 1 callers
Function
ten_crop
Crop the given PIL Image into four corners and the central crop plus the flipped version of these (horizontal flipping is used by default).
utils/transformed.py:332
↓ 1 callers
Function
test
(test_loader, epoch, Hnet, Rnet, criterion)
main.py:422
↓ 1 callers
Function
to_pil_image
Convert a tensor or an ndarray to PIL Image. See :class:`~torchvision.transforms.ToPIlImage` for more details. Args: pic (Tensor or nu
utils/transformed.py:90
↓ 1 callers
Function
to_tensor
Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor. See ``ToTensor`` for more details. Args: pic (PIL Image or numpy.ndarray): Ima
utils/transformed.py:41
↓ 1 callers
Function
train
(train_loader, epoch, Hnet, Rnet, criterion)
main.py:270
↓ 1 callers
Function
validation
(val_loader, epoch, Hnet, Rnet, criterion)
main.py:362
Method
__call__
(self, img)
utils/transformed.py:503
Method
__call__
Args: pic (PIL Image or numpy.ndarray): Image to be converted to tensor. Returns: Tensor: Converted image.
utils/transformed.py:515
Method
__call__
Args: pic (Tensor or numpy.ndarray): Image to be converted to PIL Image. Returns: PIL Image: Image converted
utils/transformed.py:542
Method
__call__
Args: tensor (Tensor): Tensor image of size (C, H, W) to be normalized. Returns: Tensor: Normalized Tensor im
utils/transformed.py:566
Method
__call__
Args: img (PIL Image): Image to be scaled. Returns: PIL Image: Rescaled image.
utils/transformed.py:593
Method
__call__
Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image.
utils/transformed.py:643
Method
__call__
Args: img (PIL Image): Image to be padded. Returns: PIL Image: Padded image.
utils/transformed.py:676
Method
__call__
(self, img)
utils/transformed.py:696
Method
__call__
Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image.
utils/transformed.py:737
Method
__call__
Args: img (PIL Image): Image to be flipped. Returns: PIL Image: Randomly flipped image.
utils/transformed.py:755
Method
__call__
Args: img (PIL Image): Image to be flipped. Returns: PIL Image: Randomly flipped image.
utils/transformed.py:770
Method
__call__
Args: img (PIL Image): Image to be flipped. Returns: PIL Image: Randomly cropped and resize image.
utils/transformed.py:828
Method
__call__
(self, img)
utils/transformed.py:868
Method
__call__
(self, img)
utils/transformed.py:893
Method
__call__
Args: tensor (Tensor): Tensor image of size (C, H, W) to be whitened. Returns: Tensor: Transformed image.
utils/transformed.py:917
Method
__call__
Args: img (PIL Image): Input image. Returns: PIL Image: Color jittered image.
utils/transformed.py:983
Method
__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
data/ImageFolderDataset.py:108
Method
__init__
(self)
main.py:513
Method
__init__
(self, transforms)
utils/transformed.py:500
Method
__init__
(self, mode=None)
utils/transformed.py:539
Method
__init__
(self, mean, std)
utils/transformed.py:562
Method
__init__
(self, size, interpolation=Image.BILINEAR)
utils/transformed.py:588
Method
__init__
(self, *args, **kwargs)
utils/transformed.py:608
Method
__init__
(self, size)
utils/transformed.py:622
Method
__init__
(self, lambd)
utils/transformed.py:692
Method
__init__
(self, size, padding=0)
utils/transformed.py:712
Method
__init__
(self, size, interpolation=Image.BILINEAR)
utils/transformed.py:793
Method
__init__
(self, *args, **kwargs)
utils/transformed.py:844
Method
__init__
(self, size)
utils/transformed.py:860
Method
__init__
(self, size, vertical_flip=False)
utils/transformed.py:884
Method
__init__
(self, transformation_matrix)
utils/transformed.py:911
Method
__init__
(self, brightness=0, contrast=0, saturation=0, hue=0)
utils/transformed.py:947
Method
__init__
(self, dir ,transform=None)
data/ImageFolderDataset.py:98
Method
__init__
(self, outer_nc, inner_nc, input_nc=None,submodule=None, outermost=False, innermost=False, norm_layer=nn.Batch
models/HidingUNet.py:42
Method
__init__
(self, nc=3, nhf=64, output_function=nn.Sigmoid)
models/RevealNet.py:16
Method
__len__
(self)
data/ImageFolderDataset.py:123
Function
adjust_gamma
Perform gamma correction on an image. Also known as Power Law Transform. Intensities in RGB mode are adjusted based on the following equation:
utils/transformed.py:459
Function
default_loader
(path)
data/ImageFolderDataset.py:65
Function
find_classes
(dir)
data/ImageFolderDataset.py:23
Method
forward
(self, input)
models/HidingUNet.py:34
Method
forward
(self, x)
models/HidingUNet.py:89
Method
forward
(self, input)
models/RevealNet.py:39
Function
make_dataset
(dir, class_to_idx)
data/ImageFolderDataset.py:30
Function
scale
(*args, **kwargs)
utils/transformed.py:202
Function
weights_init
(m)
main.py:86