From 17154dbb223cf11897c173a4e3c6fdc771d137e9 Mon Sep 17 00:00:00 2001 From: Tim Laibacher <tim.laibacher@idiap.ch> Date: Mon, 12 Aug 2019 16:42:02 +0200 Subject: [PATCH] Remove pdfcreator --- bob/ip/binseg/script/binseg.py | 1 - bob/ip/binseg/utils/rsttable.py | 20 +++++++++++++++++++- 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/bob/ip/binseg/script/binseg.py b/bob/ip/binseg/script/binseg.py index 44166070..b68b8a0c 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 1db04e45..bab97608 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' -- GitLab