-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathconfig.py
71 lines (60 loc) · 2.13 KB
/
config.py
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#! /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 = "./data/classes/coco.names"
# __C.YOLO.CLASSES = "./data/classes/voc.names"
#类别名
# __C.YOLO.CLASSES = "./data/classes/plate.names"
__C.YOLO.CLASSES = "./plate.names"
__C.YOLO.ANCHORS = "./data/anchors/yolo_anchors_gray.txt"
__C.YOLO.MOVING_AVE_DECAY = 0.9995
__C.YOLO.STRIDES = [8, 16, 32]
__C.YOLO.ANCHOR_PER_SCALE = 3
__C.YOLO.IOU_LOSS_THRESH = 0.5
__C.YOLO.UPSAMPLE_METHOD = "resize"
__C.YOLO.ORIGINAL_WEIGHT = "./checkpoint/yolov3_coco.ckpt"
__C.YOLO.DEMO_WEIGHT = "./checkpoint/yolov3_coco_demo.ckpt"
# Train options
__C.TRAIN = edict()
#__C.TRAIN.ANNOT_PATH = "./data/dataset/voc_train.txt"
__C.TRAIN.ANNOT_PATH = "./data/dataset/datasetnew_gray_train.txt"
# __C.TRAIN.BATCH_SIZE = 6
__C.TRAIN.BATCH_SIZE = 2
__C.TRAIN.INPUT_SIZE = [256, 288, 320, 352, 384, 416, 448, 480, 512, 544, 576, 608]
__C.TRAIN.DATA_AUG = True
__C.TRAIN.LEARN_RATE_INIT = 1e-4
__C.TRAIN.LEARN_RATE_END = 1e-6
__C.TRAIN.WARMUP_EPOCHS = 2
__C.TRAIN.FISRT_STAGE_EPOCHS = 20
__C.TRAIN.SECOND_STAGE_EPOCHS = 30
__C.TRAIN.INITIAL_WEIGHT = "./checkpoint/yolov3_coco_demo.ckpt"
#__C.TRAIN.INITIAL_WEIGHT = "./checkpoint/yolov3_test1_loss=0.5687.ckpt-1"
# TEST options
__C.TEST = edict()
__C.TEST.ANNOT_PATH = "./data/dataset/datasetnew_gray_test.txt"
__C.TEST.BATCH_SIZE = 2
__C.TEST.INPUT_SIZE = 544
__C.TEST.DATA_AUG = False
__C.TEST.WRITE_IMAGE = True
__C.TEST.WRITE_IMAGE_PATH = "./data/detection/"
__C.TEST.WRITE_IMAGE_SHOW_LABEL = False
#__C.TEST.WEIGHT_FILE = "./checkpoint/yolov3_test_loss=9.2099.ckpt-5"
__C.TEST.SHOW_LABEL = False
__C.TEST.SCORE_THRESHOLD = 0.3
__C.TEST.IOU_THRESHOLD = 0.45