diff --git a/bob/ip/binseg/script/binseg.py b/bob/ip/binseg/script/binseg.py index 44166070d13eeea92d77b6b54c4e1b49e62241c5..b68b8a0c64e2161b3174a4311d8d7ee78cbc9746 100644 --- a/bob/ip/binseg/script/binseg.py +++ b/bob/ip/binseg/script/binseg.py @@ -28,7 +28,6 @@ from bob.ip.binseg.engine.ssltrainer import do_ssltrain from bob.ip.binseg.engine.inferencer import do_inference from bob.ip.binseg.utils.plot import plot_overview from bob.ip.binseg.utils.click import OptionEatAll -from bob.ip.binseg.utils.pdfcreator import create_pdf, get_paths from bob.ip.binseg.utils.rsttable import create_overview_grid from bob.ip.binseg.utils.plot import metricsviz, overlay,savetransformedtest from bob.ip.binseg.utils.transformfolder import transformfolder as transfld diff --git a/bob/ip/binseg/utils/rsttable.py b/bob/ip/binseg/utils/rsttable.py index 1db04e45892c5993e711529cb6ebc4b4ea41a43a..bab9760843b759bde291430f9d6469d391e7df77 100644 --- a/bob/ip/binseg/utils/rsttable.py +++ b/bob/ip/binseg/utils/rsttable.py @@ -1,8 +1,26 @@ -from bob.ip.binseg.utils.pdfcreator import get_paths import pandas as pd from tabulate import tabulate import os +def get_paths(output_path, filename): + """ + Parameters + ---------- + output_path : str + path in which to look for files + filename : str + + Returns + ------- + list + list of file paths + """ + datadir = Path(output_path) + file_paths = sorted(list(datadir.glob('**/{}'.format(filename)))) + file_paths = [f.as_posix() for f in file_paths] + return file_paths + + def create_overview_grid(output_path): """ Reads all Metrics.csv in a certain output path and pivots them to a rst grid table""" filename = 'Metrics.csv'