📸 [Add] a inference example

.gitignore

@@ -42,6 +42,7 @@ htmlcov/
 .coverage
 .coverage.*
 .cache
+*.cache
 nosetests.xml
 coverage.xml
 *.cover
@@ -140,3 +141,6 @@ runs
 
 # Ignore npm packages (if using frontend components)
 node_modules/
+
+# Not ignore image for demo
+!demo/images/inference/*

examples/lazy.py

@@ -0,0 +1,37 @@
+import sys
+from pathlib import Path
+
+import hydra
+import torch
+
+project_root = Path(__file__).resolve().parent.parent
+sys.path.append(str(project_root))
+
+from yolo.config.config import Config
+from yolo.model.yolo import get_model
+from yolo.tools.data_loader import create_dataloader
+from yolo.tools.solver import ModelTester, ModelTrainer
+from yolo.utils.logging_utils import custom_logger, validate_log_directory
+
+
+@hydra.main(config_path="../yolo/config", config_name="config", version_base=None)
+def main(cfg: Config):
+    custom_logger()
+
+    custom_logger()
+    save_path = validate_log_directory(cfg, cfg.name)
+    dataloader = create_dataloader(cfg)
+    device = torch.device(cfg.device)
+    model = get_model(cfg).to(device)
+
+    if cfg.task.task == "train":
+        trainer = ModelTrainer(cfg, model, save_path, device)
+        trainer.solve(dataloader)
+
+    if cfg.task.task == "inference":
+        tester = ModelTester(cfg, model, save_path, device)
+        tester.solve(dataloader)
+
+
+if __name__ == "__main__":
+    main()

yolo/config/task/inference.yaml

@@ -7,5 +7,5 @@ data:
   pin_memory: True
   data_augment: {}
 nms:
-  min_confidence: 0.5
+  min_confidence: 0.75
   min_iou: 0.5