-
Notifications
You must be signed in to change notification settings - Fork 72
/
defaults_template.py
32 lines (29 loc) · 1.15 KB
/
defaults_template.py
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
# Paths
# Fill this according to own setup
BACKGROUND_DIR = ''
BACKGROUND_GLOB_STRING = '*/*.png'
POISSON_BLENDING_DIR = '' # if downloaded from S
SELECTED_LIST_FILE = 'selected.txt'
DISTRACTOR_LIST_FILE = 'neg_list.txt'
DISTRACTOR_DIR = ''
DISTRACTOR_GLOB_STRING = '*.jpg'
INVERTED_MASK = False # Set to true if white pixels represent background
# Parameters for generator
NUMBER_OF_WORKERS = 1#26
BLENDING_LIST = ['gaussian','poisson', 'none', 'box', 'motion']
# Parameters for images
MIN_NO_OF_OBJECTS = 3
MAX_NO_OF_OBJECTS = 8
MIN_NO_OF_DISTRACTOR_OBJECTS = 2
MAX_NO_OF_DISTRACTOR_OBJECTS = 4
WIDTH = 640
HEIGHT = 480
MAX_ATTEMPTS_TO_SYNTHESIZE = 20
# Parameters for objects in images
MIN_SCALE = 0.15 # min scale for scale augmentation
MAX_SCALE = 0.6 # max scale for scale augmentation
MAX_DEGREES = 30 # max rotation allowed during rotation augmentation
MAX_TRUNCATION_FRACTION = 0.25 # max fraction to be truncated = MAX_TRUNCACTION_FRACTION*(WIDTH/HEIGHT)
MAX_ALLOWED_IOU = 0.75 # IOU > MAX_ALLOWED_IOU is considered an occlusion
MIN_WIDTH = 6 # Minimum width of object to use for data generation
MIN_HEIGHT = 6 # Minimum height of object to use for data generation