diff --git a/thermo_scenes/scripts/preprocess_thermal.py b/thermo_scenes/scripts/preprocess_thermal.py index e3c100c..64f2b67 100644 --- a/thermo_scenes/scripts/preprocess_thermal.py +++ b/thermo_scenes/scripts/preprocess_thermal.py @@ -16,8 +16,8 @@ class Paths: """Path to the output rgb images""" path_to_csv_files: str """Path to the output csv files of temperature values""" - path_to_txt: str = "temperature_bounds.json" - """Path to the txt file with temperature bounds for the dataset""" + path_to_json: str + """Path to the output json file with temperature bounds for the dataset""" def main() -> None: @@ -37,7 +37,9 @@ def main() -> None: flir.save_normalised_thermal_images( paths.path_to_thermal_images_curated, paths.path_to_csv_files ) - flir.save_temperature_bounds(paths.path_to_txt) + flir.save_temperature_bounds( + str(Path(paths.path_to_json, "temperature_bounds.json")) + ) if __name__ == "__main__":