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;