final code cleanup, closes #2

This commit is contained in:
Florian Förster 2025-10-23 11:24:39 +02:00
parent 9dc7b4f6ff
commit e5529dae21
2 changed files with 5 additions and 12 deletions

View File

@ -5,17 +5,10 @@ LIB_ROOT_PATH: Final[Path] = Path(__file__).parent
STOP_FOLDER_NAME: Final[str] = "python" STOP_FOLDER_NAME: Final[str] = "python"
MODEL_FOLDER_NAME: Final[str] = "models" MODEL_FOLDER_NAME: Final[str] = "models"
# TODO: remove comment THRESHOLD_BW: Final[int] = 63
THRESHOLD_BW: Final[int] = 63 # threshold to distringuish black (electrodes) and white areas BACKBONE: Final[str] = "resnet18"
# model_path = [
# r"C:\Users\demon\Documents\EKF\Modelle\patchcore_model_links.pth",
# r"C:\Users\demon\Documents\EKF\Modelle\patchcore_model_rechts.pth",
# ] # path to anomaly detection models
BACKBONE: Final[str] = "resnet18" # parameters for AI model
LAYERS: Final[tuple[str, str]] = ("layer1", "layer2") LAYERS: Final[tuple[str, str]] = ("layer1", "layer2")
RATIO: Final[float] = 0.05 RATIO: Final[float] = 0.05
NUM_VALID_ELECTRODES: Final[int] = 6 NUM_VALID_ELECTRODES: Final[int] = 6
# TODO: Remove?
# CONTOUR_EXPORT_FILENAME_SUFFIX: Final[str] = "_all_contours"
HEATMAP_FILENAME_SUFFIX: Final[str] = "_Heatmap" HEATMAP_FILENAME_SUFFIX: Final[str] = "_Heatmap"

View File

@ -1,10 +1,10 @@
class ImageNotReadError(Exception): class ImageNotReadError(Exception):
"""thrown if image was not read successfully""" """image was not read successfully"""
class ContourCalculationError(Exception): class ContourCalculationError(Exception):
"""thrown if contour detection was not successful""" """contour detection was not successful"""
class InvalidElectrodeCount(Exception): class InvalidElectrodeCount(Exception):
"""thrown if the number of electrodes does not match the expected value""" """the number of electrodes does not match the expected value"""