Commit 904a9a15 authored by pbethge's avatar pbethge
Browse files

use 7 lang model as standard

parent 3753ff02
......@@ -20,10 +20,25 @@
typedef std::map<int, std::string> Labels;
static Labels labelsMap = {
{0, "noise"},
{1, "english"},
{2, "french"},
{3, "german"},
{4, "spanish"},
};
#define USE_MODEL_V2
#ifdef USE_MODEL_V1
static Labels labelsMap = {
{0, "noise"},
{1, "english"},
{2, "french"},
{3, "german"},
{4, "spanish"}
};
#else
static Labels labelsMap = {
{0, "noise"},
{1, "chinese"},
{2, "english"},
{3, "french"},
{4, "german"},
{5, "italian"},
{6, "russian"}
{7, "spanish"},
};
#endif
\ No newline at end of file
......@@ -26,8 +26,11 @@ void ofApp::setup() {
ofBackground(54, 54, 54);
// load the model, bail out on error
//std::string modelName = "model_4lang"; // model v1
std::string modelName = "model_attrnn"; // model v2
#ifdef USE_MODEL_V1
std::string modelName = "model_4lang"; // model v1
#else
std::string modelName = "model_7lang"; // model v2
#endif
if(!model.load(modelName)) {
std::exit(EXIT_FAILURE);
}
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment