DenseMammogram/utils.py

import shutil
import os
from os.path import join


class AverageMeter:
    """Computes and stores the average and current value"""
    def __init__(self):
        self.reset()

    def reset(self):
        self.val = 0
        self.avg = 0
        self.sum = 0
        self.count = 0

    def update(self, val, n=1):
        self.val = val
        self.sum += val * n
        self.count += n
        self.avg = self.sum / self.count

def create_backup(folders = None, files = None, backup_dir = 'experiments'):
    if folders is None:
        folders = ['.', 'corr_lists','detection']
    if files is None:
        files = ['.py', '.txt', '.json','.cfg']

    for folder in folders:
        if not os.path.isdir(folder):
            continue
        for file in os.listdir(folder):
            if file.endswith(tuple(files)):
                if folder != '.':
                    src = join(folder, file)
                    dest = join(backup_dir, folder, file)
                else:
                    src = file
                    dest = join(backup_dir, file)
                os.makedirs(os.path.split(dest)[0], exist_ok=True)
                shutil.copy(src, dest)