coco.yml 1.1 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 32 33 34 35 36 37 38 39 40 41 42 43 44
              TO_BGR: False
            - OP: ArrangeRCNN
        BATCH_SIZE: 1
        IS_PADDING: True
        DROP_LAST: False
        
    VAL:
        OPS:
            - OP: DecodeImage
              TO_RGB: True
            - OP: ResizeImage
              TARGET_SIZE: 224
            - OP: ArrangeSSD
        BATCH_SIZE: 1
    WORKER_CONF:
        BUFSIZE: 200
        WORKER_NUM: 8
        USE_PROCESS: False