mdx.mtmc.utils.io_utils module

class ValidateFile(option_strings, dest, nargs=None, const=None, default=None, type=None, choices=None, required=False, help=None, metavar=None)

Bases: Action

Module to validate files

load_csv_from_file(file_path: str) List[List[str]]

Loads from a CSV file

Parameters:

file_path (str) – file path

Returns:

data in the file

Return type:

List[List[str]]

data = load_csv_from_file(file_path)
load_csv_to_dataframe_from_file(file_path: str, column_names: List[str]) pandas.DataFrame

Loads dataframe from a CSV file

Parameters:
  • file_path (str) – file path

  • column_names (List[str]) – column names

Returns:

dataframe in the file

Return type:

pd.DataFrame

dataFrame = load_csv_to_dataframe_from_file(file_path, column_names)
load_json_from_file(file_path: str) Any

Loads JSON data from a file

Parameters:

file_path (str) – file path

Returns:

data in the file

Return type:

Any

data = load_json_from_file(file_path)
load_json_from_file_line_by_line(file_path: str) List[Any]

Loads JSON data from a file line by line

Parameters:

file_path (str) – file path

Returns:

list of data in the file

Return type:

List[Any]

data = load_json_from_file_line_by_line(file_path)
load_json_from_str(string: str) Any

Loads JSON data from a string

Parameters:

string (str) – string

Returns:

data in the string

Return type:

Any

data = load_json_from_str(string)
make_clean_dir(dir_path: str) None

Makes a clean directory

Parameters:

dir_path (str) – directory path

Returns:

None

make_clean_dir(dir_path)
make_dir(dir_path: str) None

Makes a directory without removing other files

Parameters:

dir_path (str) – directory path

Returns:

None

make_dir(dir_path)
make_seq_ini_file(gt_dir: str, camera: str, seq_length: int) None

Makes a sequence-ini file used by TrackEval library

Parameters:
  • gt_dir (str) – file path

  • camera (str) – Name of a single sensor

  • seq_length (int) – Number of frames

Returns:

None

make_seq_ini_file(gt_dir, camera, seq_length)
make_seq_maps_file(file_dir: str, sensor_ids: Set[str], benchmark: str, split_to_eval: str) None

Makes a sequence-maps file used by TrackEval library

Parameters:
  • file_dir (str) – file path

  • sensor_ids (Set(str)) – names of sensors

  • benchmark (str) – name of the benchmark

  • split_to_eval (str) – name of the split of data

Returns:

None

make_seq_maps_file(file_dir, sensor_ids)
move_files(source_dir: str, destination_dir: str) None

Moves contents of a folder from source_dir to destination_dir (non-recursive)

Parameters:
  • source_dir (str) – source directory path

  • destination_dir (str) – destination directory path

Returns:

None

move_files(source_dir, destination_dir)
remove_dir(dir_path: str) None

Removes a directory

Parameters:

dir_path (str) – directory path

Returns:

None

remove_dir(dir_path)
sanitize_string(input_string: str) str

Sanitizes an input string

Parameters:

input_string (str) – input string

Returns:

sanitized string

Return type:

str

sanitized_string = sanitize_string(input_string)
validate_file_path(input_string: str) str

Validates whether the input string matches a file path pattern

Parameters:

input_string (str) – input string

Returns:

validated file path

Return type:

str

file_path = validate_file_path(input_string)
write_dataframe_to_csv_file(file_path: str, data: pandas.DataFrame, delimiter: str = ' ') None

Writes dataframe to a CSV file

Parameters:
  • file_path (str) – file path

  • data (pd.DataFrame) – dataframe to be written

  • delimiter (str) – delimiter of the CSV file

Returns:

None

write_dataframe_to_csv_file(file_path, data, delimiter)
write_pydantic_list_to_file(file_path: str, data: Any) None

Writes a list of pydantic data to a file

Parameters:
  • file_path (str) – file path

  • data (Any) – data to be written

Returns:

None

write_pydantic_list_to_file(file_path, data)