click2mask / scripts /constants.py
omeregev's picture
Initial commit
6df18f5
class Const:
HW = 512
H = 512
W = 512
LATENT_SIZE = (64, 64)
BATCH_SIZE = 1
NUM_INFERENCE_STEPS = 100 # Number of inference steps in mask evolution
NUM_STATIC_INFERENCE_STEPS = 100 # Number of inference steps in final runs
BLENDING_START_PERCENTAGE = 0.25
DYN_START = 0.2
DYN_COND_STOP = 0.25
DYN_FINAL_STOP = 0.3
N_OUTS_FOR_DYN_MASK = 3
N_RUNS_ON_SCORES = 2
N_AUGS_ON_SCORES = 32
AUG_NUM = 2
POTENTIAL_PEAK = 0.08
SIGMA_FOR_SHAPE64 = 10.0
BIAS_DILATION_VAL = 0.6
BIAS_DILATION_DEC_FACTOR = 0.94
ADDITION_IN_COLLAPSE = 0.3
POINT_ON_LATENT_RADIUS = 1.7
OUT_RING_WIDTH = 3
IN_ON_RING_WIDTH = 6
GUIDANCE_SCALE = 14.0
THRESH_VAL = 0
MASK_LR = 4.0
RERUN_MAX_AREA_RATIO_FOR_DILATION = 0.2
MAX_AREA_RATIO_FOR_DILATION = 0.02
ALPHA_MASK_DILATION_ON_512 = 40
RERUN_STOP_DILATION = 4
GAUSS_K_GRADS = 3
GAUSS_SIGMA_GRADS = 1
GAUSS_K_MASK = 5
GAUSS_SIGMA_MASK = 1
GAUSS_K_THRESHED = 7
GAUSS_SIGMA_THRESHED = 3
THRESH_POST_GAUSS = 0.45
CLOSE_GAPS_WITH_CONNECTION_THICKNESS = 3
CLOSING_K = 3
BACK_PRES_SIGMA = 25
BACK_PRES_SCALE = 20
MAX_SEED = 2**32 - 1
class N:
POTENTIAL = "potential"
LATENT_MASK = "latent_mask"