e1250 commited on
Commit
37c15c7
·
1 Parent(s): 83180e3
Files changed (1) hide show
  1. 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):