|
7 | 7 | # File name : config.py
|
8 | 8 | # Author : YunYang1994
|
9 | 9 | # Created date: 2019-02-28 13:06:54
|
10 |
| -# Description : |
| 10 | +# Description : configurations |
11 | 11 | #
|
12 | 12 | #================================================================
|
13 | 13 |
|
14 | 14 | from easydict import EasyDict as edict
|
15 | 15 | import os
|
16 | 16 |
|
| 17 | +GIT_ROOT = "C:\\Users\\developer\\Anaconda_3\\rdt-reader" |
| 18 | + |
| 19 | +print("Git root is: "+GIT_ROOT) |
| 20 | + |
17 | 21 | __C = edict()
|
18 | 22 | # Consumers can get config by: from config import cfg
|
19 | 23 |
|
|
23 | 27 | __C.YOLO = edict()
|
24 | 28 |
|
25 | 29 | # Set the class name
|
26 |
| -__C.YOLO.CLASSES = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/classes/rdt.names" |
27 |
| -__C.YOLO.ANCHORS = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/anchors/rdt_anchors.txt" |
| 30 | +__C.YOLO.CLASSES = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/classes/rdt.names" |
| 31 | +__C.YOLO.ANCHORS = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/anchors/rdt_anchors.txt" |
28 | 32 | __C.YOLO.MOVING_AVE_DECAY = 0.9995
|
29 | 33 | __C.YOLO.STRIDES = [8, 16, 32]
|
30 | 34 | __C.YOLO.ANCHOR_PER_SCALE = 3
|
31 | 35 | __C.YOLO.IOU_LOSS_THRESH = 0.7
|
32 | 36 | __C.YOLO.UPSAMPLE_METHOD = "resize"
|
33 |
| -__C.YOLO.ORIGINAL_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/checkpoint/yolov3_coco.ckpt" |
34 |
| -__C.YOLO.DEMO_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/checkpoint/yolov3_coco_demo.ckpt" |
| 37 | +__C.YOLO.ORIGINAL_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/yolov3_coco.ckpt" |
| 38 | +__C.YOLO.DEMO_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/yolov3_coco_demo.ckpt" |
35 | 39 |
|
36 | 40 | # Train options
|
37 | 41 | __C.TRAIN = edict()
|
38 | 42 |
|
39 |
| -__C.TRAIN.ANNOT_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" |
| 43 | +__C.TRAIN.ANNOT_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" |
40 | 44 | __C.TRAIN.BATCH_SIZE = 1
|
41 | 45 | __C.TRAIN.INPUT_SIZE = [320, 352, 384, 416, 448, 480, 512, 544, 576, 608]
|
42 | 46 | __C.TRAIN.DATA_AUG = True
|
|
45 | 49 | __C.TRAIN.WARMUP_EPOCHS = 2
|
46 | 50 | __C.TRAIN.FISRT_STAGE_EPOCHS = 1
|
47 | 51 | __C.TRAIN.SECOND_STAGE_EPOCHS = 10
|
48 |
| -__C.TRAIN.INITIAL_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" |
| 52 | +__C.TRAIN.INITIAL_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" |
49 | 53 |
|
50 | 54 |
|
51 | 55 |
|
52 | 56 | # TEST options
|
53 | 57 | __C.TEST = edict()
|
54 | 58 |
|
55 |
| -__C.TEST.ANNOT_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" |
| 59 | +__C.TEST.ANNOT_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" |
56 | 60 | __C.TEST.BATCH_SIZE = 2
|
57 | 61 | __C.TEST.INPUT_SIZE = 512
|
58 | 62 | __C.TEST.DATA_AUG = False
|
59 | 63 | __C.TEST.WRITE_IMAGE = True
|
60 |
| -__C.TEST.WRITE_IMAGE_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/detection" |
| 64 | +__C.TEST.WRITE_IMAGE_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/detection" |
61 | 65 | __C.TEST.WRITE_IMAGE_SHOW_LABEL = True
|
62 |
| -__C.TEST.WEIGHT_FILE = "D:/source/repos/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" |
| 66 | +__C.TEST.WEIGHT_FILE = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" |
63 | 67 | __C.TEST.SHOW_LABEL = True
|
64 | 68 | __C.TEST.SCORE_THRESHOLD = 0.3
|
65 | 69 | __C.TEST.IOU_THRESHOLD = 0.3
|
|
0 commit comments