Spaces:
Running
Running
File size: 954 Bytes
a383d0e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
class Config:
def __init__(self):
# cnn 4 classes
# self.MODEL_PATH = 'E:/Mulong/Model/ui_compos/cnn6_icon.h5' # cnn 4 classes
# self.class_map = ['Image', 'Icon', 'Button', 'Input']
# resnet 14 classes
# self.DATA_PATH = "E:/Mulong/Datasets/rico/elements-14-2"
# self.MODEL_PATH = 'E:/Mulong/Model/rico_compos/resnet-ele14.h5'
# self.class_map = ['Button', 'CheckBox', 'Chronometer', 'EditText', 'ImageButton', 'ImageView',
# 'ProgressBar', 'RadioButton', 'RatingBar', 'SeekBar', 'Spinner', 'Switch',
# 'ToggleButton', 'VideoView', 'TextView'] # ele-14
self.DATA_PATH = "E:\Mulong\Datasets\dataset_webpage\Components3"
self.MODEL_PATH = 'E:/Mulong/Model/rico_compos/cnn2-textview.h5'
self.class_map = ['Text', 'Non-Text']
self.image_shape = (32, 32, 3)
self.class_number = len(self.class_map)
|