deepface/extension/constants.py
2020-10-19 23:58:40 +11:00

85 lines
2.3 KiB
Python

CONSTANTS = {
'ANIMAL_TYPE': ['cat', 'dog'],
'EMOTION': {
'HAPPY': 'Happy',
'NEUTRAL': 'Neutral',
'ANXIOUS': 'Anxious',
'SAD': 'Sad',
'UNSETTLED': 'Unsettled'
},
'STATISTICAL_DATA': {
"Cat": {
"feedback_number": 0,
"prediction_number": 0,
"prediction_data": {
"breed": {},
"emotion": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
}
},
"feedback_data": {
"breed": {
"wrong": {},
"correct": {}
},
"emotion": {
"wrong": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
},
"correct": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
}
}
}
},
"Dog": {
"feedback_number": 0,
"prediction_number": 0,
"prediction_data": {
"breed": {},
"emotion": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
}
},
"feedback_data": {
"breed": {
"wrong": {},
"correct": {}
},
"emotion": {
"wrong": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
},
"correct": {
"Happy": 0,
"Neutral": 0,
"Anxious": 0,
"Sad": 0,
"Unsettled": 0
}
}
}
}
}
}