rcnn_dataset.yml 857 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
DATA:
    TRAIN:
        ANNO_FILE: data/coco.test/train2017.roidb
        IMAGE_DIR: data/coco.test/train2017
        SAMPLES: 10
        IS_SHUFFLE: True
        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
22
            - OP: Permute
23 24 25 26 27
              TO_BGR: False
            - OP: ArrangeRCNN
        BATCH_SIZE: 1
        IS_PADDING: True
        DROP_LAST: False
W
walloollaw 已提交
28 29 30 31 32
        WORKER_CONF:
            BUFSIZE: 100
            WORKER_NUM: 4
            MEMSIZE: 2G
            USE_PROCESS: True