From 97bd0bdfd8016e37fe237f73b94bb3c89c76ed45 Mon Sep 17 00:00:00 2001 From: Anand Date: Thu, 14 Nov 2019 18:21:15 +0530 Subject: [PATCH] update config with folders for docker --- core/config.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/core/config.py b/core/config.py index 26d7ec2cf..057462035 100644 --- a/core/config.py +++ b/core/config.py @@ -7,13 +7,14 @@ # File name : config.py # Author : YunYang1994 # Created date: 2019-02-28 13:06:54 -# Description : +# Description : configurations # #================================================================ from easydict import EasyDict as edict import os + __C = edict() # Consumers can get config by: from config import cfg @@ -23,20 +24,20 @@ __C.YOLO = edict() # Set the class name -__C.YOLO.CLASSES = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/classes/rdt.names" -__C.YOLO.ANCHORS = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/anchors/rdt_anchors.txt" +__C.YOLO.CLASSES = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/classes/rdt.names" +__C.YOLO.ANCHORS = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/anchors/rdt_anchors.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.7 __C.YOLO.UPSAMPLE_METHOD = "resize" -__C.YOLO.ORIGINAL_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/checkpoint/yolov3_coco.ckpt" -__C.YOLO.DEMO_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/checkpoint/yolov3_coco_demo.ckpt" +__C.YOLO.ORIGINAL_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/yolov3_coco.ckpt" +__C.YOLO.DEMO_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/yolov3_coco_demo.ckpt" # Train options __C.TRAIN = edict() -__C.TRAIN.ANNOT_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" +__C.TRAIN.ANNOT_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" __C.TRAIN.BATCH_SIZE = 1 __C.TRAIN.INPUT_SIZE = [320, 352, 384, 416, 448, 480, 512, 544, 576, 608] __C.TRAIN.DATA_AUG = True @@ -45,21 +46,21 @@ __C.TRAIN.WARMUP_EPOCHS = 2 __C.TRAIN.FISRT_STAGE_EPOCHS = 1 __C.TRAIN.SECOND_STAGE_EPOCHS = 10 -__C.TRAIN.INITIAL_WEIGHT = "D:/source/repos/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" +__C.TRAIN.INITIAL_WEIGHT = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" # TEST options __C.TEST = edict() -__C.TEST.ANNOT_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" +__C.TEST.ANNOT_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/dataset/rdt_train.txt" __C.TEST.BATCH_SIZE = 2 __C.TEST.INPUT_SIZE = 512 __C.TEST.DATA_AUG = False __C.TEST.WRITE_IMAGE = True -__C.TEST.WRITE_IMAGE_PATH = "D:/source/repos/rdt-reader/tensorflow-yolov3/data/detection" +__C.TEST.WRITE_IMAGE_PATH = "/home/rdtreader/rdt-reader/tensorflow-yolov3/data/detection" __C.TEST.WRITE_IMAGE_SHOW_LABEL = True -__C.TEST.WEIGHT_FILE = "D:/source/repos/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" +__C.TEST.WEIGHT_FILE = "/home/rdtreader/rdt-reader/tensorflow-yolov3/checkpoint/rdt_model" __C.TEST.SHOW_LABEL = True __C.TEST.SCORE_THRESHOLD = 0.3 __C.TEST.IOU_THRESHOLD = 0.3