Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/akanimax/pro_gan_pytorch-examples
/ functions
Functions
17 in github.com/akanimax/pro_gan_pytorch-examples
⨍
Functions
17
◇
Types & classes
2
↓ 1 callers
Method
__setup_files
private helper for setting up the files_list :return: files => list of paths of files
implementation/data_processing/DataLoader.py:11
↓ 1 callers
Method
__setup_files
private helper for setting up the files_list :return: files => list of paths of files
implementation/data_processing/DataLoader.py:82
↓ 1 callers
Function
get_config
parse and load the provided configuration :param conf_file: configuration file :return: conf => parsed configuration
implementation/train_network.py:46
↓ 1 callers
Function
main
Main function for the script :param args: parsed command line arguments :return: None
implementation/generate_loss_plots.py:72
↓ 1 callers
Function
main
Main function for the script :param args: parsed command line arguments :return: None
implementation/train_network.py:61
↓ 1 callers
Function
parse_arguments
command line arguments parser :return: args => parsed command line arguments
implementation/generate_loss_plots.py:54
↓ 1 callers
Function
parse_arguments
command line arguments parser :return: args => parsed command line arguments
implementation/train_network.py:20
↓ 1 callers
Function
plot_loss
plot the discriminator loss values and save the plot if required :param loss_vals: (Variable Arg) numpy array or Sequence like for plotting v
implementation/generate_loss_plots.py:20
↓ 1 callers
Function
read_loss_log
read and load the loss values from a loss.log file :param file_name: path of the loss.log file :param delimiter: delimiter used to delimi
implementation/generate_loss_plots.py:8
Method
__getitem__
obtain the image (read and transform) :param idx: index of the file required :return: img => image array
implementation/data_processing/DataLoader.py:47
Method
__getitem__
obtain the image (read and transform) :param idx: index of the file required :return: img => image array
implementation/data_processing/DataLoader.py:122
Method
__init__
constructor for the class :param data_dir: path to the directory containing the data :param transform: transforms to be appli
implementation/data_processing/DataLoader.py:27
Method
__init__
constructor for the class :param data_dir: path to the directory containing the data :param transform: transforms to be appli
implementation/data_processing/DataLoader.py:102
Method
__len__
compute the length of the dataset :return: len => length of dataset
implementation/data_processing/DataLoader.py:40
Method
__len__
compute the length of the dataset :return: len => length of dataset
implementation/data_processing/DataLoader.py:115
Function
get_data_loader
generate the data_loader from the given dataset :param dataset: PyTorch dataset :param batch_size: batch size of the data :param num_
implementation/data_processing/DataLoader.py:174
Function
get_transform
obtain the image transforms required for the input data :param new_size: size of the resized images :return: image_transform => transform
implementation/data_processing/DataLoader.py:151