-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
31 lines (26 loc) · 852 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
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
# from tensorflow.compat.v1 import ConfigProto
# from tensorflow.compat.v1 import InteractiveSession
# config = ConfigProto()
# session = InteractiveSession(config=config)
# parameters
CHECKPOINT_FILE = './service_weights.h5'
MODEL_SAVE = 'service_type_model.h5'
FAST_RUN = False
IMAGE_WIDTH = 300
IMAGE_HEIGHT = 300
IMAGE_SIZE = (IMAGE_WIDTH, IMAGE_HEIGHT)
IMAGE_CHANNELS = 3
IMG_DIR = 'dataset/imagens/'
BATCH_SIZE = 32
NUM_CLASSES = 4
EPOCHS=50
DATASET = "dataset/imagens/"
MASKON_FOLDER = DATASET + "maskon"
MASKOFF_FOLDER = DATASET + "maskoff"
MASKCHIN_FOLDER = DATASET + "maskchin"
MASKMOUTH_FOLDER = DATASET + "maskmouth"
CAFFE_PROTO_FILE = "caffe_face_detector/deploy.prototxt.txt"
MODEL_FILE = "caffe_face_detector/res10_300x300_ssd_iter_140000.caffemodel"
CONFIDENCE = 0.7