_C.VIS_THRESH = 0.7 # pretained model config _C.HAND_MODEL = os.path.join(PROJECT_ROOT,'weights','det_hand.pth') _C.KEY_WHITE_MODEL = './weights/experments/network/record/white_conv3net_add_network.pth' _C.KEY_BLACK_MODEL = './weights/experments/network/record/black_conv3net_add_network.pth' _C.MEAN = [0.45734706, 0.43338275, 0.40058118] _C.STD = [0.23965294, 0.23532275, 0.2398498] ## key_config _C.KEY_NUM_CLASSES = 2 _C.WHITE_KEY_THRESH = 0.6 _C.BLACK_KEY_THRESH = 0.4 _C.NEAR_KEY_THRESH = 0.95 _C.WHITE_INPUT_CHANNEL = 1 _C.BLACK_INPUT_CHANNEL = 1 _C.WHITE_INPUT_SIZE = [112,32] _C.BLACK_INPUT_SIZE = [112,32] ##keyboard config _C.KEYBOARD_MODEL = os.path.join(PROJECT_ROOT,'weights','keyboard.pth') _C.KEYBOARD_PALETTE = [0,0,0,64,0,128] _C.KEYBOARD_NUM_CLASSES = 2 ## save file config if not EXPERMENT_CONFIG['on']: _C.SAVE_IMG_DIR = os.path.join('/home/data/lj/Piano','saved','network') else: if EXPERMENT_CONFIG['alpha']: _C.SAVE_IMG_DIR = os.path.join('/home/data/lj/Piano','saved','experment','alpha','alpha3.0') elif EXPERMENT_CONFIG['network']: