# -*- coding: utf-8 -*-
"""
author:LTH
data:
"""


# 以voc2012数据集为训练标准

# 全局变量配置类
class GetConfig:
    def __init__(self):
        # region 项目配置基本信息
        self.project_name = "yolov4 目标检测"
        self.use_cuda = True
        self.seed = 123
        # endregion

        self.target_dir = "voc2012"

        # region 数据集和dataloader配置基本信息
        self.annotations = "./data/" + self.target_dir + "/annotations.txt"
        self.val_num = 0.05
        self.image_size = 416  # 让图片输入分辨率为416 * 416，这一步的目的是为了让后面产生的卷积特征图宽高都为奇数，这样就可以产生一个center cell
        self.train_batch_size = 4
        self.test_batch_size = 1
        # endregion

        # region 模型初始化配置
        self.resume = True
        self.pretrained_weight = "./weight/best.pth"
        self.t_weight = "./t/best.pth"
        # endregion

        # region 模型训练过程配置
        self.lr = 2e-4
        self.epochs = 200
        # endregion

        # region 其他配置
        self.classes_path = "./data/" + self.target_dir + "/class.txt"
        self.anchor_path = "./data/" + self.target_dir + "/anchors.txt"
        # endregion