baseline.py 文件源码

def calculate(arguments):
    filenames = [os.path.join(arguments.input_path, filename) for filename in os.listdir(arguments.input_path) if filename.endswith(arguments.ext)]
    medians = []
    for filename in filenames:
        depth_map = read_file(filename)
        median = np.nanmedian(depth_map)
        medians.append(median)

    medians = np.array(medians)

    # Check and create output directory.
    if not os.path.exists(arguments.output_path):
        os.makedirs(arguments.output_path)

    with open(os.path.join(arguments.output_path, "median.txt"), "w") as file:
        file.write("{:.6f}".format(np.median(medians)))
        print("{:.6f}".format(np.median(medians)))