Defining image extensions only in one place
This commit is contained in:
@@ -7,6 +7,7 @@ import yaml
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, Optional
|
||||
from src.utils.logger import get_logger
|
||||
from src.utils.image import Image
|
||||
|
||||
|
||||
logger = get_logger(__name__)
|
||||
@@ -46,14 +47,7 @@ class ConfigManager:
|
||||
"database": {"path": "data/detections.db"},
|
||||
"image_repository": {
|
||||
"base_path": "",
|
||||
"allowed_extensions": [
|
||||
".jpg",
|
||||
".jpeg",
|
||||
".png",
|
||||
".tif",
|
||||
".tiff",
|
||||
".bmp",
|
||||
],
|
||||
"allowed_extensions": Image.SUPPORTED_EXTENSIONS,
|
||||
},
|
||||
"models": {
|
||||
"default_base_model": "yolov8s-seg.pt",
|
||||
@@ -232,5 +226,5 @@ class ConfigManager:
|
||||
def get_allowed_extensions(self) -> list:
|
||||
"""Get list of allowed image file extensions."""
|
||||
return self.get(
|
||||
"image_repository.allowed_extensions", [".jpg", ".jpeg", ".png"]
|
||||
"image_repository.allowed_extensions", Image.SUPPORTED_EXTENSIONS
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user