Пример #1
0
_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']: