Spaces:
Sleeping
Sleeping
fix:
Browse files- config/settings.py +4 -4
config/settings.py
CHANGED
|
@@ -18,12 +18,12 @@ yaml.SafeLoader.add_constructor("!join", join_tag)
|
|
| 18 |
class PathsConfig(BaseModel):
|
| 19 |
"""Contains paths of directories"""
|
| 20 |
project_dir: str
|
| 21 |
-
models_dir: str
|
| 22 |
# logs_dir: str
|
| 23 |
|
| 24 |
class YoloConfig(BaseModel):
|
| 25 |
"""Contains yolo configurations"""
|
| 26 |
-
model_path: str
|
| 27 |
classes: List[str]
|
| 28 |
batch_size: int
|
| 29 |
epochs: int
|
|
@@ -33,12 +33,12 @@ class YoloConfig(BaseModel):
|
|
| 33 |
|
| 34 |
class SecurityDetector(BaseModel):
|
| 35 |
"Contains Security Detectors like Smoke - Fire"
|
| 36 |
-
model_path: str
|
| 37 |
classes: List[str]
|
| 38 |
|
| 39 |
class DepthConfig(BaseModel):
|
| 40 |
"Contains depths estimation configurations"
|
| 41 |
-
model_path: str
|
| 42 |
encoder: Literal["vits", "vitb", "vitl", "vitg"]
|
| 43 |
|
| 44 |
class IntervalsConfig(BaseModel):
|
|
|
|
| 18 |
class PathsConfig(BaseModel):
|
| 19 |
"""Contains paths of directories"""
|
| 20 |
project_dir: str
|
| 21 |
+
# models_dir: str
|
| 22 |
# logs_dir: str
|
| 23 |
|
| 24 |
class YoloConfig(BaseModel):
|
| 25 |
"""Contains yolo configurations"""
|
| 26 |
+
# model_path: str
|
| 27 |
classes: List[str]
|
| 28 |
batch_size: int
|
| 29 |
epochs: int
|
|
|
|
| 33 |
|
| 34 |
class SecurityDetector(BaseModel):
|
| 35 |
"Contains Security Detectors like Smoke - Fire"
|
| 36 |
+
# model_path: str
|
| 37 |
classes: List[str]
|
| 38 |
|
| 39 |
class DepthConfig(BaseModel):
|
| 40 |
"Contains depths estimation configurations"
|
| 41 |
+
# model_path: str
|
| 42 |
encoder: Literal["vits", "vitb", "vitl", "vitg"]
|
| 43 |
|
| 44 |
class IntervalsConfig(BaseModel):
|