CV-1-目标检测-03-RCNN-05-Config

# -- encoding:utf-8 --


IMAGE_WIDTH = 227
IMAGE_HEIGHT = 227
IMAGE_CHANNEL = 3
CLASS_NUMBER = 3

ALEX_NET_MAT_FILE_PATH = "../AlexNet预加载模型/imagenet-caffe-alex.mat"

ORIGINAL_FINE_TUNE_DATA_FILE_PATH = '../rcnn_datas/fine_tune_list.txt'
TRAIN_DATA_FILE_PATH = '../rcnn_datas/rcnn/traning_data.npy'
TRAIN_LABEL_DICT_FILE_PATH = '../rcnn_datas/rcnn/label_dict.pkl'

FINE_TUNE_SUMMARY_WRITER_LOG_DIR = './output/graph/fine_tune'
FINE_TUNE_CHECKPOINT_DIR = './output/models/fine_tune'
FINE_TUNE_CHECKPOINT_FILENAME = 'models.ckpt'
FINE_TUNE_MAX_STEP = 10000
FINE_TUNE_SUMMARY_STEP = 10
FINE_TUNE_CHECKPOINT_STEP = 600
FINE_TUNE_INITIAL_LEARNING_RATE = 0.001
FINE_TUNE_DECAY_STEPS = 1000
FINE_TUNE_DECAY_RATE = 0.9
FINE_TUNE_IOU_THRESHOLD = 0.5
FINE_TUNE_POSITIVE_BATCH_SIZE = 8*3
FINE_TUNE_NEGATIVE_BATCH_SIZE = 24*3

HIGHER_FEATURES_BATCH_SIZE = 100
HIGHER_FEATURES_IOU_THRESHOLD = 0.3

TRAIN_SVM_HIGHER_FEATURES_DATA_FILE_PATH = "../rcnn_datas/svm/higher_features_{}.npy"
SVM_CHECKPOINT_FILE_PATH = "./output/models/svm/model_{}.pkl"
SVM_PREDICT_PROBABILITY_THRESHOLD = 0.7
SVM_PREDICT_PROBABILITY_WITH_NMS = True
SVM_PREDICT_PROBABILITY_WITH_NMS_IOU_THRESHOLD = 0.5

TRAIN_REGRESSION_HIGHER_FEATURES_DATA_FILE_PATH = '../rcnn_datas/regression/higher_features.npy'
REGRESSION_BOX_IOU_THRESHOLD = 0.6
REGRESSION_BOX_BATCH_SIZE = 64
REGRESSION_INPUT_DIMENSION = 9216
REGRESSION_OUTPUT_DIMENSION = 4
REGRESSION_SUMMARY_WRITER_LOG_DIR = './output/graph/regression'
REGRESSION_CHECKPOINT_DIR = './output/models/regression'
REGRESSION_CHECKPOINT_FILENAME = 'models.ckpt'
REGRESSION_MAX_STEP = 10000
REGRESSION_SUMMARY_STEP = 10
REGRESSION_CHECKPOINT_STEP = 500
REGRESSION_INITIAL_LEARNING_RATE = 0.0001
REGRESSION_DECAY_STEPS = 1000
REGRESSION_DECAY_RATE = 0.9
REGRESSION_BATCH_SIZE = 64