File size: 475 Bytes
b80dc1e 642e32f 5fed6e3 787b81d b80dc1e 5fed6e3 787b81d 5fed6e3 787b81d b80dc1e 642e32f 5fed6e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import argparse
from loguru import logger
from model.yolo import get_model
from utils.tools import load_model_cfg, custom_logger
import hydra
from config.config import Config
from omegaconf import OmegaConf
@hydra.main(config_path="config", config_name="config", version_base=None)
def main(cfg: Config):
OmegaConf.set_struct(cfg, False)
model = get_model(cfg.model)
logger.info("Success load model")
if __name__ == "__main__":
custom_logger()
main()
|