detection_rcnn.yaml 533 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
DEPLOY:
    USE_GPU: 1
    MODEL_PATH: "/root/projects/models/faster_rcnn_pp50"
    MODEL_FILENAME: "__model__"
    PARAMS_FILENAME: "__params__"
    EVAL_CROP_SIZE: (608, 608)
    RESIZE_TYPE: "RANGE_SCALING"
    TARGET_SHORT_SIZE : 800
    MEAN:  [0.485, 0.456, 0.406]
    STD: [0.229, 0.224, 0.225]
    IMAGE_TYPE: "rgb"
    NUM_CLASSES: 1
    CHANNELS : 3
    PRE_PROCESSOR: "DetectionPreProcessor"
    PREDICTOR_MODE: "ANALYSIS"
B
Bin Long 已提交
16
    BATCH_SIZE : 1
17 18
    RESIZE_MAX_SIZE: 1333
    FEEDS_SIZE: 3
B
Bin Long 已提交
19 20
    USE_TRT : 0
    TRT_MODE : FP32