📸 [Add] a inference example
Browse files- .gitignore +4 -0
- demo/images/inference/image.png +0 -0
- examples/lazy.py +37 -0
- yolo/config/task/inference.yaml +1 -1
.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.
|
11 |
min_iou: 0.5
|
|
|
7 |
pin_memory: True
|
8 |
data_augment: {}
|
9 |
nms:
|
10 |
+
min_confidence: 0.75
|
11 |
min_iou: 0.5
|