params.NUM_SHAPE = 6 params.NUM_CLASSES = 4 # (3 shapes + 1 background) params.NUM_FILTER = [256, 128] params.ETH_MEAN_SHIFT_THRESHOLD = 1.5 params.DELTA_VAR = 0.5 params.DELTA_D = 1.5 params.IMG_SIZE = 256 params.OUTPUT_SIZE = 64 params.SEQUENCE_LEN = 100 params.TRAIN_NUM_SEQ = 500 params.VAL_NUM_SEQ = 20 params.TEST_NUM_SEQ = 20 params.RANDOM_SIZE = True params.OPTICAL_FLOW_WEIGHT = 0 params.BACKBONE = 'xception' params.GITHUB_DIR = 'C:/Users/yliu60/Documents/GitHub' params.LEARNING_RATE = 1e-4 params.EPOCHS = 30 params.EPOCHS_PER_SAVE = 5 params.STEPS_PER_VISUAL = 1000 params.FEATURE_STRING = f'{params.NUM_SHAPE}_shapes' params.MODEL_SAVE_DIR = f'model/{params.NUM_SHAPE}_shapes' params.TRAIN_SET_PATH = f'dataset/{params.NUM_SHAPE}_shapes/train' params.VAL_SET_PATH = f'dataset/{params.NUM_SHAPE}_shapes/val' params.TEST_SET_PATH = f'dataset/{params.NUM_SHAPE}_shapes/test' params.IOU_THRESHOLD = 0.5 params.MASK_AREA_THRESHOLD = 20 experiment = Experiment(params) experiment.init_model() experiment.epoch = 0