-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.py
43 lines (28 loc) · 919 Bytes
/
config.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
33
34
35
36
37
38
39
40
41
42
43
import os
class Directories:
# generate this many training images
num_generated_images = 12000
# size of images
# input and output images will be resized to be these sizes
image_width = 640
image_height = 480
image_depth = 3
segmented = 0
num_iters_per_fg_and_bg = 1
def __init__(self):
self.root_dir = os.path.abspath(os.path.dirname(__file__))
# image directories
self.input_image_directory = self.root_dir + "/images/samples/input/"
self.background_image_directory = self.root_dir + "/images/samples/background/"
# save dirs
self.save_image_dir = self.root_dir + "/images/generated/images/"
self.save_anno_dir = self.root_dir + "/images/generated/anno/"
# labels for extracted objects
class Labels:
thumb = "thumb"
index = "index"
middle = "middle"
ring = "ring"
pinky = "pinky"
def get_labels(self):
return [self.thumb, self.index, self.middle, self.ring, self.pinky]