Compare commits

..

2 Commits

View File

@ -24,7 +24,7 @@ class ColourSpace(Enum):
BGR = 0 BGR = 0
HSV = 1 HSV = 1
class colourTester: class ColourTester:
def __init__(self) -> None: def __init__(self) -> None:
self.bgr_curves = [[[], [], []] for x in range(BARK_TYPES)] self.bgr_curves = [[[], [], []] for x in range(BARK_TYPES)]
self.bgr_ids = ["", "", ""] self.bgr_ids = ["", "", ""]
@ -154,7 +154,7 @@ class colourTester:
return return
if __name__ == "__main__": if __name__ == "__main__":
tester = colourTester() tester = ColourTester()
tester.setCsvBgrPath(CSV_PATH + "bgr_curve.csv") tester.setCsvBgrPath(CSV_PATH + "bgr_curve.csv")
tester.setCsvHsvPath(CSV_PATH + "hsv_curve.csv") tester.setCsvHsvPath(CSV_PATH + "hsv_curve.csv")
tester.testDataset("dataset") tester.testDataset("dataset")