Module monk.pytorch.transforms.retrieve_transform

Expand source code
from pytorch.transforms.imports import *
from pytorch.transforms.common import set_transforms
from system.imports import *




def retrieve_trainval_transforms(system_dict):
    '''
    Retrieve training and validdation transforms in copy-from, and resume states

    Args:
        system_dict (dict): System dictionary storing experiment state and set variables

    Returns:
        dict: updated system dict
    '''
    set_phases = ["train", "val"];
    system_dict = set_transforms(system_dict, set_phases);
    return system_dict;




def retrieve_test_transforms(system_dict):
    '''
    Retrieve testing transforms in copy-from, and resume states

    Args:
        system_dict (dict): System dictionary storing experiment state and set variables

    Returns:
        dict: updated system dict
    '''
    set_phases = ["test"];
    system_dict= set_transforms(system_dict, set_phases);
    return system_dict;

Functions

def retrieve_test_transforms(system_dict)

Retrieve testing transforms in copy-from, and resume states

Args

system_dict : dict
System dictionary storing experiment state and set variables

Returns

dict
updated system dict
Expand source code
def retrieve_test_transforms(system_dict):
    '''
    Retrieve testing transforms in copy-from, and resume states

    Args:
        system_dict (dict): System dictionary storing experiment state and set variables

    Returns:
        dict: updated system dict
    '''
    set_phases = ["test"];
    system_dict= set_transforms(system_dict, set_phases);
    return system_dict;
def retrieve_trainval_transforms(system_dict)

Retrieve training and validdation transforms in copy-from, and resume states

Args

system_dict : dict
System dictionary storing experiment state and set variables

Returns

dict
updated system dict
Expand source code
def retrieve_trainval_transforms(system_dict):
    '''
    Retrieve training and validdation transforms in copy-from, and resume states

    Args:
        system_dict (dict): System dictionary storing experiment state and set variables

    Returns:
        dict: updated system dict
    '''
    set_phases = ["train", "val"];
    system_dict = set_transforms(system_dict, set_phases);
    return system_dict;