henry000 commited on
Commit
7100919
·
1 Parent(s): 597f02f

📸 [Add] a inference example

Browse files
.gitignore CHANGED
@@ -42,6 +42,7 @@ htmlcov/
42
  .coverage
43
  .coverage.*
44
  .cache
 
45
  nosetests.xml
46
  coverage.xml
47
  *.cover
@@ -140,3 +141,6 @@ runs
140
 
141
  # Ignore npm packages (if using frontend components)
142
  node_modules/
 
 
 
 
42
  .coverage
43
  .coverage.*
44
  .cache
45
+ *.cache
46
  nosetests.xml
47
  coverage.xml
48
  *.cover
 
141
 
142
  # Ignore npm packages (if using frontend components)
143
  node_modules/
144
+
145
+ # Not ignore image for demo
146
+ !demo/images/inference/*
demo/images/inference/image.png ADDED
examples/lazy.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from pathlib import Path
3
+
4
+ import hydra
5
+ import torch
6
+
7
+ project_root = Path(__file__).resolve().parent.parent
8
+ sys.path.append(str(project_root))
9
+
10
+ from yolo.config.config import Config
11
+ from yolo.model.yolo import get_model
12
+ from yolo.tools.data_loader import create_dataloader
13
+ from yolo.tools.solver import ModelTester, ModelTrainer
14
+ from yolo.utils.logging_utils import custom_logger, validate_log_directory
15
+
16
+
17
+ @hydra.main(config_path="../yolo/config", config_name="config", version_base=None)
18
+ def main(cfg: Config):
19
+ custom_logger()
20
+
21
+ custom_logger()
22
+ save_path = validate_log_directory(cfg, cfg.name)
23
+ dataloader = create_dataloader(cfg)
24
+ device = torch.device(cfg.device)
25
+ model = get_model(cfg).to(device)
26
+
27
+ if cfg.task.task == "train":
28
+ trainer = ModelTrainer(cfg, model, save_path, device)
29
+ trainer.solve(dataloader)
30
+
31
+ if cfg.task.task == "inference":
32
+ tester = ModelTester(cfg, model, save_path, device)
33
+ tester.solve(dataloader)
34
+
35
+
36
+ if __name__ == "__main__":
37
+ main()
yolo/config/task/inference.yaml CHANGED
@@ -7,5 +7,5 @@ data:
7
  pin_memory: True
8
  data_augment: {}
9
  nms:
10
- min_confidence: 0.5
11
  min_iou: 0.5
 
7
  pin_memory: True
8
  data_augment: {}
9
  nms:
10
+ min_confidence: 0.75
11
  min_iou: 0.5