File size: 831 Bytes
8f0b970 a33e03b 8f0b970 a33e03b 8f0b970 a33e03b 8f0b970 7330b76 8f0b970 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
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 create_model
from yolo.tools.data_loader import create_dataloader
from yolo.tools.solver import ModelTester
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()
save_path = validate_log_directory(cfg, cfg.name)
dataloader = create_dataloader(cfg)
device = torch.device(cfg.device)
model = create_model(cfg).to(device)
tester = ModelTester(cfg, model, save_path, device)
tester.solve(dataloader)
if __name__ == "__main__":
main()
|