coco.yml 1.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
DATA:
    TRAIN:
        ANNO_FILE: data/coco.test/train2017.roidb
        IMAGE_DIR: data/coco.test/train2017
        SAMPLES: 10
        TYPE: RoiDbSource
    VAL: 
        ANNO_FILE: data/coco.test/val2017.roidb
        IMAGE_DIR: data/coco.test/val2017
        SAMPLES: 10
        TYPE: RoiDbSource
TRANSFORM:
    TRAIN:
        OPS:
            - OP: DecodeImage
              TO_RGB: False
            - OP: RandomFlipImage
              PROB: 0.5
            - OP: NormalizeImage
              MEAN: [102.9801, 115.9465, 122.7717]
              IS_SCALE: False
              IS_CHANNEL_FIRST: False
            - OP: ResizeImage
              TARGET_SIZE: 800
              MAX_SIZE: 1333
26
            - OP: Permute
27 28 29 30 31
              TO_BGR: False
            - OP: ArrangeRCNN
        BATCH_SIZE: 1
        IS_PADDING: True
        DROP_LAST: False
W
walloollaw 已提交
32 33 34 35 36
        WORKER_CONF:
            BUFSIZE: 100
            WORKER_NUM: 4
            USE_PROCESS: True
            MEMSIZE: 2G
37 38 39 40 41 42 43 44
    VAL:
        OPS:
            - OP: DecodeImage
              TO_RGB: True
            - OP: ResizeImage
              TARGET_SIZE: 224
            - OP: ArrangeSSD
        BATCH_SIZE: 1
W
walloollaw 已提交
45 46 47 48
        WORKER_CONF:
            BUFSIZE: 100
            WORKER_NUM: 4
            USE_PROCESS: True