Skip to content

Commit da99dff

Browse files
committed
2 parents 1724a17 + b665f50 commit da99dff

File tree

1 file changed

+14
-10
lines changed

1 file changed

+14
-10
lines changed

core/config.py

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,17 @@
77
# File name : config.py
88
# Author : YunYang1994
99
# Created date: 2019-02-28 13:06:54
10-
# Description :
10+
# Description : configurations
1111
#
1212
#================================================================
1313

1414
from easydict import EasyDict as edict
1515
import os
1616

17+
GIT_ROOT = "C:\\Users\\developer\\Anaconda_3\\rdt-reader"
18+
19+
print("Git root is: "+GIT_ROOT)
20+
1721
__C = edict()
1822
# Consumers can get config by: from config import cfg
1923

@@ -23,20 +27,20 @@
2327
__C.YOLO = edict()
2428

2529
# 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"
2832
__C.YOLO.MOVING_AVE_DECAY = 0.9995
2933
__C.YOLO.STRIDES = [8, 16, 32]
3034
__C.YOLO.ANCHOR_PER_SCALE = 3
3135
__C.YOLO.IOU_LOSS_THRESH = 0.7
3236
__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"
3539

3640
# Train options
3741
__C.TRAIN = edict()
3842

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"
4044
__C.TRAIN.BATCH_SIZE = 1
4145
__C.TRAIN.INPUT_SIZE = [320, 352, 384, 416, 448, 480, 512, 544, 576, 608]
4246
__C.TRAIN.DATA_AUG = True
@@ -45,21 +49,21 @@
4549
__C.TRAIN.WARMUP_EPOCHS = 2
4650
__C.TRAIN.FISRT_STAGE_EPOCHS = 1
4751
__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"
4953

5054

5155

5256
# TEST options
5357
__C.TEST = edict()
5458

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"
5660
__C.TEST.BATCH_SIZE = 2
5761
__C.TEST.INPUT_SIZE = 512
5862
__C.TEST.DATA_AUG = False
5963
__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"
6165
__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"
6367
__C.TEST.SHOW_LABEL = True
6468
__C.TEST.SCORE_THRESHOLD = 0.3
6569
__C.TEST.IOU_THRESHOLD = 0.3

0 commit comments

Comments
 (0)