diff --git a/src/experiments/algorithms/colour_FE.py b/src/experiments/algorithms/colour_FE.py index b0bdb4a..5d38210 100644 --- a/src/experiments/algorithms/colour_FE.py +++ b/src/experiments/algorithms/colour_FE.py @@ -24,7 +24,7 @@ class ColourSpace(Enum): BGR = 0 HSV = 1 -class colourTester: +class ColourTester: def __init__(self) -> None: self.bgr_curves = [[[], [], []] for x in range(BARK_TYPES)] self.bgr_ids = ["", "", ""] @@ -154,7 +154,7 @@ class colourTester: return if __name__ == "__main__": - tester = colourTester() + tester = ColourTester() tester.setCsvBgrPath(CSV_PATH + "bgr_curve.csv") tester.setCsvHsvPath(CSV_PATH + "hsv_curve.csv") tester.testDataset("dataset")