config.py

#! /usr/bin/env python
# coding=utf-8
#================================================================
#   Copyright (C) 2019 * Ltd. All rights reserved.
#
#   Editor      : VIM
#   File name   : config.py
#   Author      : YunYang1994
#   Created date: 2019-02-28 13:06:54
#   Description :
#
#================================================================

from easydict import EasyDict as edict


__C                           = edict()
# Consumers can get config by: from config import cfg

cfg                           = __C

# YOLO options
__C.YOLO                      = edict()

# Set the class name
__C.YOLO.CLASSES              = "/home/thura/Desktop/gear_teeth_detection/classes/gear_teeth.names"
__C.YOLO.ANCHORS              = "/home/thura/Desktop/gear_teeth_detection/anchors/basline_anchors.txt"
__C.YOLO.STRIDES              = [8, 16, 32]
__C.YOLO.ANCHOR_PER_SCALE     = 3
__C.YOLO.IOU_LOSS_THRESH      = 0.5

# Train options
__C.TRAIN                     = edict()

__C.TRAIN.ANNOT_PATH          = "/home/thura/Desktop/gear_teeth_detection/dataset/train_gear_annotations.txt"
__C.TRAIN.BATCH_SIZE          = 2
# __C.TRAIN.INPUT_SIZE            = [320, 352, 384, 416, 448, 480, 512, 544, 576, 608]
# __C.TRAIN.INPUT_SIZE          = [416]
__C.TRAIN.INPUT_SIZE          = [256]
__C.TRAIN.DATA_AUG            = True
__C.TRAIN.LR_INIT             = 1e-5
__C.TRAIN.LR_END              = 1e-6
__C.TRAIN.WARMUP_EPOCHS       = 2
__C.TRAIN.EPOCHS              = 100


# TEST options
__C.TEST                      = edict()

__C.TEST.ANNOT_PATH           = "/home/thura/Desktop/gear_teeth_detection/dataset/test_gear_annotations.txt"
__C.TEST.BATCH_SIZE           = 1
__C.TEST.INPUT_SIZE           = 256
__C.TEST.DATA_AUG             = False
__C.TEST.DECTECTED_IMAGE_PATH = "/home/thura/Desktop/gear_teeth_detection/result_output/eval_detection/"
__C.TEST.SCORE_THRESHOLD      = 0.3
__C.TEST.IOU_THRESHOLD        = 0.5