| import os | |
| # Пути | |
| BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | |
| MODEL_PATH = os.path.join(BASE_DIR, "weights", "iter5_3.pt") | |
| # Классы | |
| NUM_CLASSES = 3 | |
| CLASS_NAMES = {0: 'Roots', 1: 'Stems', 2: 'Leaves'} | |
| COLORS = {0: (0, 0, 255), 1: (0, 255, 255), 2: (0, 255, 0)} | |
| # Настройки инференса по умолчанию | |
| TILE_SIZE = 512 | |
| TARGET_SIZE = 640 | |
| STRIDE = 256 | |
| CONF_THRESH = 0.3 | |
| MASK_THRESH = 0.15 | |
| # Настройки анализа | |
| MORPH_CLOSING_RADIUS = 3 | |
| MIN_OBJECT_SIZE = 150 | |
| RADIUS_CORRECTION_PX = 2 | |
| MIN_RADIUS = 0.5 | |
| PIXELS_PER_MM = 9.6 |