File size: 3,984 Bytes
690f890 |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
# -*- coding: utf-8 -*-
# Copyright (c) Alibaba, Inc. and its affiliates.
from easydict import EasyDict
######################### Common #########################
#------------------------ image ------------------------#
image_plain_anno = EasyDict()
image_plain_anno.NAME = "PlainImageAnnotator"
image_plain_anno.INPUTS = {"image": None}
image_plain_anno.OUTPUTS = {"image": None}
image_mask_plain_anno = EasyDict()
image_mask_plain_anno.NAME = "PlainMaskAnnotator"
image_mask_plain_anno.INPUTS = {"mask": None}
image_mask_plain_anno.OUTPUTS = {"mask": None}
image_maskaug_plain_anno = EasyDict()
image_maskaug_plain_anno.NAME = "PlainMaskAugAnnotator"
image_maskaug_plain_anno.INPUTS = {"mask": None}
image_maskaug_plain_anno.OUTPUTS = {"mask": None}
image_maskaug_invert_anno = EasyDict()
image_maskaug_invert_anno.NAME = "PlainMaskAugInvertAnnotator"
image_maskaug_invert_anno.INPUTS = {"mask": None}
image_maskaug_invert_anno.OUTPUTS = {"mask": None}
image_maskaug_anno = EasyDict()
image_maskaug_anno.NAME = "MaskAugAnnotator"
image_maskaug_anno.INPUTS = {"mask": None, 'mask_cfg': None}
image_maskaug_anno.OUTPUTS = {"mask": None}
image_mask_draw_anno = EasyDict()
image_mask_draw_anno.NAME = "MaskDrawAnnotator"
image_mask_draw_anno.INPUTS = {"mask": None, 'image': None, 'bbox': None, 'mode': None}
image_mask_draw_anno.OUTPUTS = {"mask": None}
image_maskaug_region_random_anno = EasyDict()
image_maskaug_region_random_anno.NAME = "RegionCanvasAnnotator"
image_maskaug_region_random_anno.SCALE_RANGE = [ 0.5, 1.0 ]
image_maskaug_region_random_anno.USE_AUG = True
image_maskaug_region_random_anno.INPUTS = {"mask": None, 'image': None, 'bbox': None, 'mode': None}
image_maskaug_region_random_anno.OUTPUTS = {"mask": None}
image_maskaug_region_crop_anno = EasyDict()
image_maskaug_region_crop_anno.NAME = "RegionCanvasAnnotator"
image_maskaug_region_crop_anno.SCALE_RANGE = [ 0.5, 1.0 ]
image_maskaug_region_crop_anno.USE_AUG = True
image_maskaug_region_crop_anno.USE_RESIZE = False
image_maskaug_region_crop_anno.USE_CANVAS = False
image_maskaug_region_crop_anno.INPUTS = {"mask": None, 'image': None, 'bbox': None, 'mode': None}
image_maskaug_region_crop_anno.OUTPUTS = {"mask": None}
#------------------------ video ------------------------#
video_plain_anno = EasyDict()
video_plain_anno.NAME = "PlainVideoAnnotator"
video_plain_anno.INPUTS = {"frames": None}
video_plain_anno.OUTPUTS = {"frames": None}
video_mask_plain_anno = EasyDict()
video_mask_plain_anno.NAME = "PlainMaskVideoAnnotator"
video_mask_plain_anno.INPUTS = {"masks": None}
video_mask_plain_anno.OUTPUTS = {"masks": None}
video_maskaug_plain_anno = EasyDict()
video_maskaug_plain_anno.NAME = "PlainMaskAugVideoAnnotator"
video_maskaug_plain_anno.INPUTS = {"masks": None}
video_maskaug_plain_anno.OUTPUTS = {"masks": None}
video_maskaug_invert_anno = EasyDict()
video_maskaug_invert_anno.NAME = "PlainMaskAugInvertVideoAnnotator"
video_maskaug_invert_anno.INPUTS = {"masks": None}
video_maskaug_invert_anno.OUTPUTS = {"masks": None}
video_mask_expand_anno = EasyDict()
video_mask_expand_anno.NAME = "ExpandMaskVideoAnnotator"
video_mask_expand_anno.INPUTS = {"masks": None}
video_mask_expand_anno.OUTPUTS = {"masks": None}
video_maskaug_anno = EasyDict()
video_maskaug_anno.NAME = "MaskAugAnnotator"
video_maskaug_anno.INPUTS = {"mask": None, 'mask_cfg': None}
video_maskaug_anno.OUTPUTS = {"mask": None}
video_maskaug_layout_anno = EasyDict()
video_maskaug_layout_anno.NAME = "LayoutMaskAnnotator"
video_maskaug_layout_anno.RAM_TAG_COLOR_PATH = "models/VACE-Annotators/layout/ram_tag_color_list.txt"
video_maskaug_layout_anno.USE_AUG = True
video_maskaug_layout_anno.INPUTS = {"mask": None, 'mask_cfg': None}
video_maskaug_layout_anno.OUTPUTS = {"mask": None}
#------------------------ prompt ------------------------#
prompt_plain_anno = EasyDict()
prompt_plain_anno.NAME = "PlainPromptAnnotator"
prompt_plain_anno.INPUTS = {"prompt": None}
prompt_plain_anno.OUTPUTS = {"prompt": None}
|